Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -9,8 +9,8 @@ decoder_pipeline = StableCascadeDecoderPipeline.from_pretrained("stabilityai/sta
|
|
| 9 |
prior_pipeline.enable_model_cpu_offload()
|
| 10 |
decoder_pipeline.enable_model_cpu_offload()
|
| 11 |
|
| 12 |
-
prior = prior_pipeline
|
| 13 |
-
decoder = decoder_pipeline
|
| 14 |
|
| 15 |
@spaces.GPU
|
| 16 |
def generate(prompt, negative_prompt, steps):
|
|
|
|
| 9 |
prior_pipeline.enable_model_cpu_offload()
|
| 10 |
decoder_pipeline.enable_model_cpu_offload()
|
| 11 |
|
| 12 |
+
prior = prior_pipeline.to("cuda")
|
| 13 |
+
decoder = decoder_pipeline.to("cuda")
|
| 14 |
|
| 15 |
@spaces.GPU
|
| 16 |
def generate(prompt, negative_prompt, steps):
|