zmelumian commited on
Commit
39316ac
1 Parent(s): c042515

now did black formatting

Browse files
Files changed (1) hide show
  1. 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
- height,
245
- width,
246
- args.num_frames,
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)