supersolar commited on
Commit
528826b
·
verified ·
1 Parent(s): 6d69f47

Update run.py

Browse files
Files changed (1) hide show
  1. run.py +6 -6
run.py CHANGED
@@ -155,7 +155,7 @@ if __name__ == "__main__":
155
  help="Path to the pre-trained model",
156
  )
157
  parser.add_argument(
158
- "--process-length", type=int, default=195, help="Number of frames to process"
159
  )
160
  parser.add_argument(
161
  "--cpu-offload",
@@ -165,19 +165,19 @@ if __name__ == "__main__":
165
  help="CPU offload option",
166
  )
167
  parser.add_argument(
168
- "--target-fps", type=int, default=15, help="Target FPS for the output video"
169
  ) # -1 for original fps
170
  parser.add_argument("--seed", type=int, default=42, help="Random seed")
171
  parser.add_argument(
172
- "--num-inference-steps", type=int, default=25, help="Number of inference steps"
173
  )
174
  parser.add_argument(
175
- "--guidance-scale", type=float, default=1.2, help="Guidance scale"
176
  )
177
  parser.add_argument("--window-size", type=int, default=110, help="Window size")
178
  parser.add_argument("--overlap", type=int, default=25, help="Overlap size")
179
- parser.add_argument("--max-res", type=int, default=1024, help="Maximum resolution")
180
- parser.add_argument("--save_npz", type=bool, default=True, help="Save npz file")
181
  parser.add_argument("--track_time", type=bool, default=False, help="Track time")
182
 
183
  args = parser.parse_args()
 
155
  help="Path to the pre-trained model",
156
  )
157
  parser.add_argument(
158
+ "--process-length", type=int, default=19500, help="Number of frames to process"
159
  )
160
  parser.add_argument(
161
  "--cpu-offload",
 
165
  help="CPU offload option",
166
  )
167
  parser.add_argument(
168
+ "--target-fps", type=int, default=-1, help="Target FPS for the output video"
169
  ) # -1 for original fps
170
  parser.add_argument("--seed", type=int, default=42, help="Random seed")
171
  parser.add_argument(
172
+ "--num-inference-steps", type=int, default=4, help="Number of inference steps"
173
  )
174
  parser.add_argument(
175
+ "--guidance-scale", type=float, default=1, help="Guidance scale"
176
  )
177
  parser.add_argument("--window-size", type=int, default=110, help="Window size")
178
  parser.add_argument("--overlap", type=int, default=25, help="Overlap size")
179
+ parser.add_argument("--max-res", type=int, default=512, help="Maximum resolution")
180
+ parser.add_argument("--save_npz", type=bool, default=Fals, help="Save npz file")
181
  parser.add_argument("--track_time", type=bool, default=False, help="Track time")
182
 
183
  args = parser.parse_args()