Hugging Face's logo

Spaces:
chats-bug
/
image-captioning
Runtime error

App Files Community
chats-bug commited on
Commit
245a3fa
·
1 Parent(s): 7917b37

Updated Model

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ device_map = {
19
  preprocessor_blip2_8_bit = AutoProcessor.from_pretrained("Salesforce/blip2-opt-2.7b")
20
  model_blip2_8_bit = Blip2ForConditionalGeneration.from_pretrained(
21
  "Salesforce/blip2-opt-2.7b",
22
- device_map=device_map,
23
  quantization_config=quantization_config,
24
  load_in_8bit=True
25
  )
 
19
  preprocessor_blip2_8_bit = AutoProcessor.from_pretrained("Salesforce/blip2-opt-2.7b")
20
  model_blip2_8_bit = Blip2ForConditionalGeneration.from_pretrained(
21
  "Salesforce/blip2-opt-2.7b",
22
+ device_map="auto",
23
  quantization_config=quantization_config,
24
  load_in_8bit=True
25
  )