Spaces:
Runtime error
Runtime error
chats-bug
commited on
Commit
·
826388b
1
Parent(s):
fdf0785
Removed unwanted params
Browse files
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
|
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")
|