AlyxTeam commited on
Commit
a183b63
1 Parent(s): 1de0132

feat: 依赖安装方法

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -4,6 +4,9 @@ from huggingface_hub import InferenceClient
4
  from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig
5
  import torch
6
  import subprocess
 
 
 
7
 
8
  subprocess.run("rm -rf /data-nvme/zerogpu-offload/*", env={}, shell=True)
9
  subprocess.run("pip install flash-attn --no-build-isolation", env={"FLASH_ATTENTION_SKIP_CUDA_BUILD": "TRUE"}, shell=True)
 
4
  from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig
5
  import torch
6
  import subprocess
7
+ import os
8
+
9
+ # os.system("pip install dashscope")
10
 
11
  subprocess.run("rm -rf /data-nvme/zerogpu-offload/*", env={}, shell=True)
12
  subprocess.run("pip install flash-attn --no-build-isolation", env={"FLASH_ATTENTION_SKIP_CUDA_BUILD": "TRUE"}, shell=True)