Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -1,3 +1,4 @@
|
|
|
|
1 |
import copy
|
2 |
import os
|
3 |
import random
|
@@ -53,7 +54,7 @@ def t2v_generation(prompt, resolution, watermark_wan, seed=-1):
|
|
53 |
gr.Warning(f"Warning: {e}")
|
54 |
return None, gr.Button(visible=True)
|
55 |
|
56 |
-
|
57 |
def t2v_generation_async(prompt, size, watermark_wan, seed=-1):
|
58 |
print(seed)
|
59 |
seed = seed if seed >= 0 else random.randint(0, 2147483647)
|
|
|
1 |
+
import spaces
|
2 |
import copy
|
3 |
import os
|
4 |
import random
|
|
|
54 |
gr.Warning(f"Warning: {e}")
|
55 |
return None, gr.Button(visible=True)
|
56 |
|
57 |
+
@spaces.GPU
|
58 |
def t2v_generation_async(prompt, size, watermark_wan, seed=-1):
|
59 |
print(seed)
|
60 |
seed = seed if seed >= 0 else random.randint(0, 2147483647)
|