Spaces:
Running
Running
Set num_workers to 1 manually
Browse files- translate.py +1 -0
translate.py
CHANGED
@@ -45,6 +45,7 @@ def get_dataloader(
|
|
45 |
dataset,
|
46 |
batch_size=batch_size,
|
47 |
collate_fn=data_collator,
|
|
|
48 |
)
|
49 |
|
50 |
|
|
|
45 |
dataset,
|
46 |
batch_size=batch_size,
|
47 |
collate_fn=data_collator,
|
48 |
+
num_workers=1,
|
49 |
)
|
50 |
|
51 |
|