Iker commited on
Commit
d3132fd
·
1 Parent(s): 5bd7f14

Fix find_executable_batch_size deprecation

Browse files
Files changed (1) hide show
  1. translate.py +1 -2
translate.py CHANGED
@@ -17,8 +17,7 @@ from transformers import (
17
 
18
  from dataset import DatasetReader, count_lines
19
 
20
- from accelerate import Accelerator, DistributedType
21
- from accelerate.memory_utils import find_executable_batch_size
22
 
23
 
24
  def get_dataloader(
 
17
 
18
  from dataset import DatasetReader, count_lines
19
 
20
+ from accelerate import Accelerator, DistributedType, find_executable_batch_size
 
21
 
22
 
23
  def get_dataloader(