Spaces:
Runtime error
Runtime error
Update streaming_svd_inference.py
Browse files
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=
|
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(
|