1inkusFace commited on
Commit
df8ddfc
·
verified ·
1 Parent(s): 4c0b936

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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)