chats-bug commited on
Commit
826388b
·
1 Parent(s): fdf0785

Removed unwanted params

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ device_map = {
17
 
18
  # Load the Blip2 model
19
  preprocessor_blip2_8_bit = AutoProcessor.from_pretrained("Salesforce/blip2-opt-6.7b")
20
- model_blip2_8_bit = Blip2ForConditionalGeneration.from_pretrained("Salesforce/blip2-opt-6.7b", device_map=device_map, load_in_8bit_fp32_cpu_offload=True)
21
 
22
  # Load the Blip base model
23
  # preprocessor_blip_base = AutoProcessor.from_pretrained("Salesforce/blip-image-captioning-base")
 
17
 
18
  # Load the Blip2 model
19
  preprocessor_blip2_8_bit = AutoProcessor.from_pretrained("Salesforce/blip2-opt-6.7b")
20
+ model_blip2_8_bit = Blip2ForConditionalGeneration.from_pretrained("Salesforce/blip2-opt-6.7b", device_map=device_map)
21
 
22
  # Load the Blip base model
23
  # preprocessor_blip_base = AutoProcessor.from_pretrained("Salesforce/blip-image-captioning-base")