dushuai112233 commited on
Commit
1707a69
·
verified ·
1 Parent(s): 831092e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -4,6 +4,7 @@ from datasets import load_dataset
4
  import torch
5
 
6
  def main():
 
7
  # 基础模型位置
8
  model_name = "dushuai112233/Qwen2-1.5B-Instruct"
9
  # 设备
 
4
  import torch
5
 
6
  def main():
7
+ accelerator = Accelerator()#初始化
8
  # 基础模型位置
9
  model_name = "dushuai112233/Qwen2-1.5B-Instruct"
10
  # 设备