dushuai112233 commited on
Commit
831092e
·
verified ·
1 Parent(s): 20600d9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -59,6 +59,7 @@ def main():
59
  train_dataset=train_dataset, # 训练数据集
60
  eval_dataset=val_dataset, # 验证数据集
61
  tokenizer=tokenizer, # 用于预处理的分词器
 
62
  )
63
 
64
  # Start Training
 
59
  train_dataset=train_dataset, # 训练数据集
60
  eval_dataset=val_dataset, # 验证数据集
61
  tokenizer=tokenizer, # 用于预处理的分词器
62
+ accelerator=accelerator,
63
  )
64
 
65
  # Start Training