Spaces:
Runtime error
Runtime error
Update stable_cascade.py
Browse files- stable_cascade.py +0 -2
stable_cascade.py
CHANGED
@@ -1,12 +1,10 @@
|
|
1 |
import torch, os
|
2 |
from diffusers import StableCascadeDecoderPipeline, StableCascadePriorPipeline
|
3 |
import gradio as gr
|
4 |
-
import spaces
|
5 |
|
6 |
prior = StableCascadePriorPipeline.from_pretrained("stabilityai/stable-cascade-prior", torch_dtype=torch.bfloat16).to("cuda")
|
7 |
decoder = StableCascadeDecoderPipeline.from_pretrained("stabilityai/stable-cascade", torch_dtype=torch.float16).to("cuda")
|
8 |
|
9 |
-
@spaces.GPU
|
10 |
def generate_images(
|
11 |
prompt="a photo of a girl",
|
12 |
negative_prompt="bad,ugly,deformed",
|
|
|
1 |
import torch, os
|
2 |
from diffusers import StableCascadeDecoderPipeline, StableCascadePriorPipeline
|
3 |
import gradio as gr
|
|
|
4 |
|
5 |
prior = StableCascadePriorPipeline.from_pretrained("stabilityai/stable-cascade-prior", torch_dtype=torch.bfloat16).to("cuda")
|
6 |
decoder = StableCascadeDecoderPipeline.from_pretrained("stabilityai/stable-cascade", torch_dtype=torch.float16).to("cuda")
|
7 |
|
|
|
8 |
def generate_images(
|
9 |
prompt="a photo of a girl",
|
10 |
negative_prompt="bad,ugly,deformed",
|