Resolving the errors
Browse files
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
|
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
|