Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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
|
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)
|