nicolabs/Open-Sora
1
0
Fork 0
mirror of https://github.com/hpcaitech/Open-Sora.git synced 2026-04-12 13:54:53 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

[gradio] removed time and resolution limitation (#140)

Browse source
This commit is contained in:
Frank Lee 2024-06-16 14:24:13 +08:00 committed by GitHub
parent 1771664792
commit ab75d98f92
1 changed files with 23 additions and 23 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

46
gradio/app.py
View file

@ -438,29 +438,29 @@ def run_video_inference(
seed,
sampling_steps,
cfg_scale):
if (resolution == "480p" and length == "16s") or \
(resolution == "720p" and length in ["8s", "16s"]):
gr.Warning("Generation is interrupted as the combination of 480p and 16s will lead to CUDA out of memory")
else:
return run_inference(
"Text2Video",
prompt_text,
resolution,
aspect_ratio,
length,
motion_strength,
aesthetic_score,
use_motion_strength,
use_aesthetic_score,
camera_motion,
reference_image,
refine_prompt,
fps,
num_loop,
seed,
sampling_steps,
cfg_scale
)
# if (resolution == "480p" and length == "16s") or \
# (resolution == "720p" and length in ["8s", "16s"]):
# gr.Warning("Generation is interrupted as the combination of 480p and 16s will lead to CUDA out of memory")
# else:
return run_inference(
"Text2Video",
prompt_text,
resolution,
aspect_ratio,
length,
motion_strength,
aesthetic_score,
use_motion_strength,
use_aesthetic_score,
camera_motion,
reference_image,
refine_prompt,
fps,
num_loop,
seed,
sampling_steps,
cfg_scale
)
def generate_random_prompt():