Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -65,7 +65,8 @@ pipe = StableDiffusionPipeline.from_pretrained(
|
|
65 |
# "sayakpaul/taesd-diffusers", torch_dtype=torch_dtype, use_safetensors=True)
|
66 |
pipe = pipe.to(device)
|
67 |
pipe.enable_vae_tiling()
|
68 |
-
|
|
|
69 |
def infer(
|
70 |
prompt,
|
71 |
negative_prompt,
|
|
|
65 |
# "sayakpaul/taesd-diffusers", torch_dtype=torch_dtype, use_safetensors=True)
|
66 |
pipe = pipe.to(device)
|
67 |
pipe.enable_vae_tiling()
|
68 |
+
|
69 |
+
@spaces.GPU #[uncomment to use ZeroGPU]
|
70 |
def infer(
|
71 |
prompt,
|
72 |
negative_prompt,
|