Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
e40c773
1
Parent(s):
bdb3667
Update SDK version in README and adjust GPU decorator placement in app.py
Browse files
README.md
CHANGED
@@ -4,7 +4,7 @@ emoji: 🏆
|
|
4 |
colorFrom: purple
|
5 |
colorTo: indigo
|
6 |
sdk: gradio
|
7 |
-
sdk_version: 5.
|
8 |
app_file: app.py
|
9 |
pinned: false
|
10 |
---
|
|
|
4 |
colorFrom: purple
|
5 |
colorTo: indigo
|
6 |
sdk: gradio
|
7 |
+
sdk_version: 5.23.0
|
8 |
app_file: app.py
|
9 |
pinned: false
|
10 |
---
|
app.py
CHANGED
@@ -64,6 +64,7 @@ stream = AsyncStream()
|
|
64 |
outputs_folder = './outputs/'
|
65 |
os.makedirs(outputs_folder, exist_ok=True)
|
66 |
|
|
|
67 |
@torch.no_grad()
|
68 |
def worker(input_image, prompt, n_prompt, seed, total_second_length, latent_window_size, steps, cfg, gs, rs, gpu_memory_preservation, use_teacache):
|
69 |
total_latent_sections = (total_second_length * 30) / (latent_window_size * 4)
|
@@ -270,7 +271,6 @@ def worker(input_image, prompt, n_prompt, seed, total_second_length, latent_wind
|
|
270 |
stream.output_queue.push(('end', None))
|
271 |
return
|
272 |
|
273 |
-
@spaces.GPU(duration=120)
|
274 |
def process(input_image, prompt, n_prompt, seed, total_second_length, latent_window_size, steps, cfg, gs, rs, gpu_memory_preservation, use_teacache):
|
275 |
global stream
|
276 |
assert input_image is not None, 'No input image!'
|
|
|
64 |
outputs_folder = './outputs/'
|
65 |
os.makedirs(outputs_folder, exist_ok=True)
|
66 |
|
67 |
+
@spaces.GPU(duration=120)
|
68 |
@torch.no_grad()
|
69 |
def worker(input_image, prompt, n_prompt, seed, total_second_length, latent_window_size, steps, cfg, gs, rs, gpu_memory_preservation, use_teacache):
|
70 |
total_latent_sections = (total_second_length * 30) / (latent_window_size * 4)
|
|
|
271 |
stream.output_queue.push(('end', None))
|
272 |
return
|
273 |
|
|
|
274 |
def process(input_image, prompt, n_prompt, seed, total_second_length, latent_window_size, steps, cfg, gs, rs, gpu_memory_preservation, use_teacache):
|
275 |
global stream
|
276 |
assert input_image is not None, 'No input image!'
|