Spaces:
Build error
Build 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",
|