nicolabs/Open-Sora
1
0
Fork 0
mirror of https://github.com/hpcaitech/Open-Sora.git synced 2026-04-11 21:42:26 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

updated gradio model version (#299)

Browse source
This commit is contained in:
Frank Lee 2024-04-24 22:52:36 +08:00 committed by GitHub
parent 04ddc76e4f
commit 0b6348d8d2
1 changed files with 5 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
gradio/app.py
View file

@ -23,10 +23,12 @@ import gradio as gr
MODEL_TYPES = ["v1.1"]
CONFIG_MAP = {
"v1.1": "configs/opensora-v1-1/inference/sample-ref.py",
"v1.1-stage2": "configs/opensora-v1-1/inference/sample-ref.py",
"v1.1-stage3": "configs/opensora-v1-1/inference/sample-ref.py",
}
HF_STDIT_MAP = {
"v1.1": "hpcai-tech/OpenSora-STDiT-v2-stage2",
"v1.1-stage2": "hpcai-tech/OpenSora-STDiT-v2-stage2",
"v1.1-stage3": "hpcai-tech/OpenSora-STDiT-v2-stage3",
}
RESOLUTION_MAP = {
"360p": (360, 480),
@ -249,7 +251,7 @@ def parse_args():
parser = argparse.ArgumentParser()
parser.add_argument(
"--model-type",
default="v1.1",
default="v1.1-stage3",
choices=MODEL_TYPES,
help=f"The type of model to run for the Gradio App, can only be {MODEL_TYPES}",
)