larry1129 commited on
Commit
7f83bc8
·
verified ·
1 Parent(s): eff2bce

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -0
app.py CHANGED
@@ -4,6 +4,18 @@ from peft import PeftModel
4
  import torch
5
  import os
6
 
 
 
 
 
 
 
 
 
 
 
 
 
7
 
8
  # 获取 Hugging Face 访问令牌
9
  hf_token = os.getenv("HF_API_TOKEN")
 
4
  import torch
5
  import os
6
 
7
+ pip install accelerate
8
+
9
+ pip install -i https://pypi.org/simple/ bitsandbytes
10
+
11
+ # 打印已安装的包版本以进行调试
12
+ import transformers
13
+ import bitsandbytes
14
+ import accelerate
15
+
16
+ print(f"transformers version: {transformers.__version__}")
17
+ print(f"bitsandbytes version: {bitsandbytes.__version__}")
18
+ print(f"accelerate version: {accelerate.__version__}")
19
 
20
  # 获取 Hugging Face 访问令牌
21
  hf_token = os.getenv("HF_API_TOKEN")