updated gradio options (#301)

This commit is contained in:
Frank Lee 2024-04-24 23:27:03 +08:00 committed by GitHub
parent 6770f2ee67
commit 11d1912b3c
3 changed files with 6 additions and 3 deletions

View file

@ -31,6 +31,8 @@ HF_STDIT_MAP = {
"v1.1-stage3": "hpcai-tech/OpenSora-STDiT-v2-stage3",
}
RESOLUTION_MAP = {
"144p": (144, 256),
"240p": (240, 426),
"360p": (360, 480),
"480p": (480, 858),
"720p": (720, 1280),
@ -452,8 +454,8 @@ def main():
lines=4,
)
resolution = gr.Radio(
choices=["360p", "480p", "720p", "1080p"],
value="360p",
choices=["144p", "240p", "360p", "480p", "720p", "1080p"],
value="144p",
label="Resolution",
)
length = gr.Radio(

View file

@ -1,3 +1,3 @@
xformers
git+https://github.com/hpcaitech/Open-Sora.git#egg=opensora
transformers
git+https://github.com/hpcaitech/Open-Sora.git#egg=opensora

View file

@ -14,3 +14,4 @@ tqdm
transformers
wandb
rotary_embedding_torch
pandarallel