Spaces:

Madiy
/

CanvasAI

Running on Zero

Madiy commited on 17 days ago

Commit

932ffee

verified ·

1 Parent(s): 44c5cfa

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -82,7 +82,6 @@ blip_processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning
 blip_model = BlipForConditionalGeneration.from_pretrained(
     "Salesforce/blip-image-captioning-base",
     torch_dtype=torch.float16,
-    resume_download= True
 ).to("cuda")
 print("All models loaded.")

 blip_model = BlipForConditionalGeneration.from_pretrained(
     "Salesforce/blip-image-captioning-base",
     torch_dtype=torch.float16,
 ).to("cuda")
 print("All models loaded.")