Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -172,9 +172,9 @@ captioner_2 = pipeline(model="Salesforce/blip-image-captioning-base",device='cud
|
|
172 |
#captioner_3 = pipeline(model="ford442/blip-image-to-text-large-bf16",device='cuda', task="image-to-text")
|
173 |
model5 = Blip2ForConditionalGeneration.from_pretrained("ford442/blip2-image-to-text-bf16").to('cuda')
|
174 |
processor5 = Blip2Processor.from_pretrained("ford442/blip2-image-to-text-bf16", device_map='cuda')
|
175 |
-
txt_tokenizer = AutoTokenizer.from_pretrained(checkpoint, device_map='cuda', add_prefix_space=False)
|
176 |
-
txt_tokenizer.tokenizer_legacy=False
|
177 |
-
model = Phi3ForCausalLM.from_pretrained(checkpoint).to('cuda:0')
|
178 |
#model = AutoModelForCausalLM.from_pretrained(checkpoint, device_map='cuda') #.to('cuda')
|
179 |
|
180 |
ip_model = IPAdapterXL(pipe, local_folder, ip_ckpt, device)
|
|
|
172 |
#captioner_3 = pipeline(model="ford442/blip-image-to-text-large-bf16",device='cuda', task="image-to-text")
|
173 |
model5 = Blip2ForConditionalGeneration.from_pretrained("ford442/blip2-image-to-text-bf16").to('cuda')
|
174 |
processor5 = Blip2Processor.from_pretrained("ford442/blip2-image-to-text-bf16", device_map='cuda')
|
175 |
+
#txt_tokenizer = AutoTokenizer.from_pretrained(checkpoint, device_map='cuda', add_prefix_space=False)
|
176 |
+
#txt_tokenizer.tokenizer_legacy=False
|
177 |
+
#model = Phi3ForCausalLM.from_pretrained(checkpoint).to('cuda:0')
|
178 |
#model = AutoModelForCausalLM.from_pretrained(checkpoint, device_map='cuda') #.to('cuda')
|
179 |
|
180 |
ip_model = IPAdapterXL(pipe, local_folder, ip_ckpt, device)
|