Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -18,7 +18,7 @@ model.device
|
|
18 |
model_id = 'prompthero/midjourney-v4-diffusion' #"stabilityai/stable-diffusion-2"
|
19 |
|
20 |
scheduler = EulerDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler")
|
21 |
-
pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.
|
22 |
pipe = pipe.to("cuda")
|
23 |
|
24 |
def transcribe(audio,prompt_num,user_keywords):
|
|
|
18 |
model_id = 'prompthero/midjourney-v4-diffusion' #"stabilityai/stable-diffusion-2"
|
19 |
|
20 |
scheduler = EulerDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler")
|
21 |
+
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)
|
22 |
pipe = pipe.to("cuda")
|
23 |
|
24 |
def transcribe(audio,prompt_num,user_keywords):
|