Spaces:

nightfury
/

Riffusion_real-time_image-to-music_generation

Runtime error

nightfury commited on Jan 7, 2023

Commit

79f6f72

•

1 Parent(s): ddb246b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,7 +14,7 @@ blip_model_id = "Salesforce/blip-image-captioning-base"
 pipe = StableDiffusionPipeline.from_pretrained(model_id)
 pipe = pipe.to("cpu") #cuda
-blip_model = BlipForConditionalGeneration.from_pretrained(blip_model_id, torch_dtype=torch.float16).to("cuda")
 processor = BlipProcessor.from_pretrained(blip_model_id)
 def predict(image):

 pipe = StableDiffusionPipeline.from_pretrained(model_id)
 pipe = pipe.to("cpu") #cuda
+blip_model = BlipForConditionalGeneration.from_pretrained(blip_model_id, torch_dtype=torch.float16).to("CPU") #CUDA
 processor = BlipProcessor.from_pretrained(blip_model_id)
 def predict(image):