jbilcke-hf HF Staff commited on
Commit
509112f
·
1 Parent(s): 189b70e

add missing stuff

Browse files
Files changed (1) hide show
  1. app.py +15 -3
app.py CHANGED
@@ -86,17 +86,29 @@ def create_args():
86
  args.reproduce = False
87
  args.load_key = "module"
88
 
89
- # Add missing 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
- # Add missing flow_reverse attribute (default is True based on config.py)
95
  args.flow_reverse = True
96
 
97
- # Add missing flow_solver attribute (default is "euler" based on config.py)
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...")