Spaces:
Running
on
A100
Running
on
A100
Commit
·
509112f
1
Parent(s):
189b70e
add missing stuff
Browse files
app.py
CHANGED
@@ -86,17 +86,29 @@ def create_args():
|
|
86 |
args.reproduce = False
|
87 |
args.load_key = "module"
|
88 |
|
89 |
-
#
|
90 |
args.text_projection = "single_refiner"
|
91 |
args.text_states_dim = 4096
|
92 |
args.text_states_dim_2 = 768
|
93 |
|
94 |
-
#
|
95 |
args.flow_reverse = True
|
96 |
|
97 |
-
#
|
98 |
args.flow_solver = "euler"
|
99 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
100 |
return args
|
101 |
|
102 |
logger.info("Initializing Hunyuan-GameCraft model...")
|
|
|
86 |
args.reproduce = False
|
87 |
args.load_key = "module"
|
88 |
|
89 |
+
# text encoder related attributes
|
90 |
args.text_projection = "single_refiner"
|
91 |
args.text_states_dim = 4096
|
92 |
args.text_states_dim_2 = 768
|
93 |
|
94 |
+
# default is True based on config.py
|
95 |
args.flow_reverse = True
|
96 |
|
97 |
+
# default is "euler" based on config.py
|
98 |
args.flow_solver = "euler"
|
99 |
|
100 |
+
# default is 256 based on config.py
|
101 |
+
args.rope_theta = 256
|
102 |
+
|
103 |
+
# default for HYVideo-T/2 model
|
104 |
+
args.patch_size = [1, 2, 2]
|
105 |
+
|
106 |
+
# default is True based on config.py
|
107 |
+
args.vae_tiling = True
|
108 |
+
|
109 |
+
# default is 0 based on config.py
|
110 |
+
args.ip_cfg_scale = 0.0
|
111 |
+
|
112 |
return args
|
113 |
|
114 |
logger.info("Initializing Hunyuan-GameCraft model...")
|