Spaces:
Running
on
A100
Running
on
A100
now did black formatting
Browse files- inference.py +4 -8
inference.py
CHANGED
@@ -240,14 +240,10 @@ def main():
|
|
240 |
assert height % 32 == 0, f"Height ({height}) should be divisible by 32."
|
241 |
assert width % 32 == 0, f"Width ({width}) should be divisible by 32."
|
242 |
assert (
|
243 |
-
|
244 |
-
|
245 |
-
|
246 |
-
|
247 |
-
)
|
248 |
-
in RECOMMENDED_RESOLUTIONS
|
249 |
-
or args.custom_resolution
|
250 |
-
), f"The selected resolution + num frames combination is not supported, results would be suboptimal. Supported (h,w,f) are: {RECOMMENDED_RESOLUTIONS}. Use --custom_resolution to enable working with this resolution."
|
251 |
|
252 |
# Paths for the separate mode directories
|
253 |
ckpt_dir = Path(args.ckpt_dir)
|
|
|
240 |
assert height % 32 == 0, f"Height ({height}) should be divisible by 32."
|
241 |
assert width % 32 == 0, f"Width ({width}) should be divisible by 32."
|
242 |
assert (
|
243 |
+
height,
|
244 |
+
width,
|
245 |
+
args.num_frames,
|
246 |
+
) in RECOMMENDED_RESOLUTIONS or args.custom_resolution, f"The selected resolution + num frames combination is not supported, results would be suboptimal. Supported (h,w,f) are: {RECOMMENDED_RESOLUTIONS}. Use --custom_resolution to enable working with this resolution."
|
|
|
|
|
|
|
|
|
247 |
|
248 |
# Paths for the separate mode directories
|
249 |
ckpt_dir = Path(args.ckpt_dir)
|