Upyaya commited on
Commit
3773d42
1 Parent(s): 43b8814

Resolving the errors

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ def init_model():
21
  processor = Blip2Processor.from_pretrained(preprocess_ckp)
22
 
23
  #Model
24
- model = Blip2ForConditionalGeneration.from_pretrained(base_model_ckp, load_in_8bit = True, device_map = "auto")
25
  model = PeftModel.from_pretrained(model, peft_model_ckp)
26
 
27
  #init_model_required = False
 
21
  processor = Blip2Processor.from_pretrained(preprocess_ckp)
22
 
23
  #Model
24
+ model = Blip2ForConditionalGeneration.from_pretrained(base_model_ckp)#, load_in_8bit = True, device_map = "auto")
25
  model = PeftModel.from_pretrained(model, peft_model_ckp)
26
 
27
  #init_model_required = False