Spaces:
Running
Running
Update simple_app.py
Browse files- simple_app.py +3 -0
simple_app.py
CHANGED
@@ -47,6 +47,8 @@ def infer(prompt, progress=gr.Progress(track_tqdm=True)):
|
|
47 |
"--sample_shift", "8",
|
48 |
"--sample_guide_scale", "6",
|
49 |
"--prompt", prompt,
|
|
|
|
|
50 |
"--save_file", "generated_video.mp4"
|
51 |
]
|
52 |
print("Starting video generation process...")
|
@@ -58,6 +60,7 @@ def infer(prompt, progress=gr.Progress(track_tqdm=True)):
|
|
58 |
bufsize=1)
|
59 |
|
60 |
print("Process started successfully.")
|
|
|
61 |
|
62 |
while True:
|
63 |
# Poll stdout with a 40ms timeout.
|
|
|
47 |
"--sample_shift", "8",
|
48 |
"--sample_guide_scale", "6",
|
49 |
"--prompt", prompt,
|
50 |
+
"--t5_cpu",
|
51 |
+
"--offload_model", True ,
|
52 |
"--save_file", "generated_video.mp4"
|
53 |
]
|
54 |
print("Starting video generation process...")
|
|
|
60 |
bufsize=1)
|
61 |
|
62 |
print("Process started successfully.")
|
63 |
+
print(process)
|
64 |
|
65 |
while True:
|
66 |
# Poll stdout with a 40ms timeout.
|