Spaces:

apruvd
/

Realtime_Speech_to_Image_Generator

Runtime error

apruvd commited on Jun 24

Commit

6b1e6f7

•

1 Parent(s): eb5d22a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,9 +15,9 @@ import PIL
 model = whisper.load_model("base")
 model.device
-# model_id = 'prompthero/midjourney-v4-diffusion' #"stabilityai/stable-diffusion-2"
-model_id = "TaiMingLu/diffusion-architecture"
 scheduler = EulerDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler")
 pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16) #pipe = StableDiffusionPipeline.from_pretrained(model_id , torch_dtype=torch.float16 #pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler, revision="fp16", torch_dtype=torch.float16)

 model = whisper.load_model("base")
 model.device
+model_id = 'prompthero/midjourney-v4-diffusion' #"stabilityai/stable-diffusion-2"
+# model_id = "TaiMingLu/diffusion-architecture"
 scheduler = EulerDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler")
 pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16) #pipe = StableDiffusionPipeline.from_pretrained(model_id , torch_dtype=torch.float16 #pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler, revision="fp16", torch_dtype=torch.float16)