Spaces:
Paused
Paused
Update app_torch.py
Browse files- app_torch.py +5 -2
app_torch.py
CHANGED
|
@@ -15,6 +15,7 @@ from huggingface_hub import hf_hub_download
|
|
| 15 |
from PIL import Image
|
| 16 |
import spaces
|
| 17 |
import time
|
|
|
|
| 18 |
# ---------------------------
|
| 19 |
# Environment flags (reduce fusion/compilation) — set early
|
| 20 |
# ---------------------------
|
|
@@ -216,7 +217,9 @@ def generate_output(prompt, mode, duration, width, height, steps, guidance, sche
|
|
| 216 |
|
| 217 |
|
| 218 |
|
| 219 |
-
|
|
|
|
|
|
|
| 220 |
|
| 221 |
|
| 222 |
if mode == "image":
|
|
@@ -252,7 +255,7 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Kandinsky 5.0 T2V (robust load)")
|
|
| 252 |
duration = gr.Slider(1, 10, step=1, value=5, label="Duration (s)")
|
| 253 |
width = gr.Radio([512, 768], value=768, label="Width")
|
| 254 |
height = gr.Radio([512, 768], value=512, label="Height")
|
| 255 |
-
steps = gr.Slider(4, 50, step=1, value=
|
| 256 |
guidance = gr.Slider(0.0, 20.0, step=0.5, value=8.0, label="Guidance Weight")
|
| 257 |
scheduler = gr.Slider(1.0, 10.0, step=0.5, value=5.0, label="Scheduler Scale")
|
| 258 |
btn = gr.Button("Generate", variant="primary")
|
|
|
|
| 15 |
from PIL import Image
|
| 16 |
import spaces
|
| 17 |
import time
|
| 18 |
+
import time, random
|
| 19 |
# ---------------------------
|
| 20 |
# Environment flags (reduce fusion/compilation) — set early
|
| 21 |
# ---------------------------
|
|
|
|
| 217 |
|
| 218 |
|
| 219 |
|
| 220 |
+
|
| 221 |
+
out_name = f"/tmp/{int(time.time())}_{random.randint(100,999)}.{'mp4' if mode == 'video' else 'png'}"
|
| 222 |
+
|
| 223 |
|
| 224 |
|
| 225 |
if mode == "image":
|
|
|
|
| 255 |
duration = gr.Slider(1, 10, step=1, value=5, label="Duration (s)")
|
| 256 |
width = gr.Radio([512, 768], value=768, label="Width")
|
| 257 |
height = gr.Radio([512, 768], value=512, label="Height")
|
| 258 |
+
steps = gr.Slider(4, 50, step=1, value=10, label="Sampling Steps")
|
| 259 |
guidance = gr.Slider(0.0, 20.0, step=0.5, value=8.0, label="Guidance Weight")
|
| 260 |
scheduler = gr.Slider(1.0, 10.0, step=0.5, value=5.0, label="Scheduler Scale")
|
| 261 |
btn = gr.Button("Generate", variant="primary")
|