Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -82,7 +82,7 @@ text_encoder_2.requires_grad_(False)
|
|
82 |
image_encoder.requires_grad_(False)
|
83 |
transformer.requires_grad_(False)
|
84 |
|
85 |
-
|
86 |
|
87 |
|
88 |
if not high_vram:
|
@@ -95,7 +95,7 @@ else:
|
|
95 |
image_encoder.to(gpu)
|
96 |
vae.to(gpu)
|
97 |
transformer.to(gpu)
|
98 |
-
|
99 |
stream = AsyncStream()
|
100 |
|
101 |
outputs_folder = './outputs/'
|
|
|
82 |
image_encoder.requires_grad_(False)
|
83 |
transformer.requires_grad_(False)
|
84 |
|
85 |
+
|
86 |
|
87 |
|
88 |
if not high_vram:
|
|
|
95 |
image_encoder.to(gpu)
|
96 |
vae.to(gpu)
|
97 |
transformer.to(gpu)
|
98 |
+
@spaces.GPU(duration=75)
|
99 |
stream = AsyncStream()
|
100 |
|
101 |
outputs_folder = './outputs/'
|