Spaces:
Runtime error
Runtime error
supersolar
commited on
Update run.py
Browse files
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=
|
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
|
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=
|
173 |
)
|
174 |
parser.add_argument(
|
175 |
-
"--guidance-scale", type=float, default=1
|
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=
|
180 |
-
parser.add_argument("--save_npz", type=bool, default=
|
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()
|