hpoghos commited on
Commit
2c1dddf
1 Parent(s): 1254e3c

Update streaming_svd_inference.py

Browse files
Files changed (1) hide show
  1. streaming_svd_inference.py +1 -1
streaming_svd_inference.py CHANGED
@@ -82,7 +82,7 @@ class StreamingSVD():
82
 
83
  model.load_state_dict(ckpt) # load trained model
84
  trainer = cli.trainer
85
- data_module_loader = partial(VideoDataModule, workers=2)
86
  vfi = i2v_enhance_interface.vfi_init(model.vfi)
87
 
88
  enhance_pipeline, enhance_generator = i2v_enhance_interface.i2v_enhance_init(
 
82
 
83
  model.load_state_dict(ckpt) # load trained model
84
  trainer = cli.trainer
85
+ data_module_loader = partial(VideoDataModule, workers=0)
86
  vfi = i2v_enhance_interface.vfi_init(model.vfi)
87
 
88
  enhance_pipeline, enhance_generator = i2v_enhance_interface.i2v_enhance_init(