1inkusFace commited on
Commit
0b70428
·
verified ·
1 Parent(s): 059e466

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -167,7 +167,7 @@ model5 = Blip2ForConditionalGeneration.from_pretrained("ford442/blip2-image-to-t
167
  processor5 = Blip2Processor.from_pretrained("ford442/blip2-image-to-text-bf16")
168
  txt_tokenizer = AutoTokenizer.from_pretrained(checkpoint, add_prefix_space=False)
169
  txt_tokenizer.tokenizer_legacy=False
170
- model = AutoModelForCausalLM.from_pretrained(checkpoint,attn_implementation="flash_attention_2").to('cuda')
171
 
172
  ip_model = IPAdapterXL(pipe, local_folder, ip_ckpt, device)
173
  text_encoder=CLIPTextModel.from_pretrained('ford442/RealVisXL_V5.0_BF16', subfolder='text_encoder',token=True).to(device=device, dtype=torch.bfloat16)
 
167
  processor5 = Blip2Processor.from_pretrained("ford442/blip2-image-to-text-bf16")
168
  txt_tokenizer = AutoTokenizer.from_pretrained(checkpoint, add_prefix_space=False)
169
  txt_tokenizer.tokenizer_legacy=False
170
+ model = AutoModelForCausalLM.from_pretrained(checkpoint).to('cuda')
171
 
172
  ip_model = IPAdapterXL(pipe, local_folder, ip_ckpt, device)
173
  text_encoder=CLIPTextModel.from_pretrained('ford442/RealVisXL_V5.0_BF16', subfolder='text_encoder',token=True).to(device=device, dtype=torch.bfloat16)