FlawedLLM commited on
Commit
583590b
·
verified ·
1 Parent(s): 9a19292

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -3,12 +3,13 @@
3
  import subprocesses
4
 
5
  subprocess.run(["pip", "install", "--upgrade", "pip"], check=True)
 
6
  subprocess.run(["pip", "install", "--upgrade", "transformers"], check=True)
7
  subprocess.run(["pip", "install", "--upgrade", "bitsandbytes"], check=True)
8
  subprocess.run(["pip", "install", "--upgrade", "peft"], check=True)
9
  subprocess.run(["pip", "install", "--upgrade", "xformers"], check=True)
10
  subprocess.run(["pip", "install", "--upgrade", "accelerate"], check=True)
11
- subprocess.run(["pip", "install", "--upgrade", "torch"], check=True)
12
 
13
  import os
14
  HF_TOKEN = os.environ["HF_TOKEN"]
 
3
  import subprocesses
4
 
5
  subprocess.run(["pip", "install", "--upgrade", "pip"], check=True)
6
+ subprocess.run(["pip", "install", "--upgrade", "torch"], check=True)
7
  subprocess.run(["pip", "install", "--upgrade", "transformers"], check=True)
8
  subprocess.run(["pip", "install", "--upgrade", "bitsandbytes"], check=True)
9
  subprocess.run(["pip", "install", "--upgrade", "peft"], check=True)
10
  subprocess.run(["pip", "install", "--upgrade", "xformers"], check=True)
11
  subprocess.run(["pip", "install", "--upgrade", "accelerate"], check=True)
12
+ subprocess.run(["pip", "install", "unsloth[cu121-ampere-torch230] @ git+https://github.com/unslothai/unsloth.git"], check=True)
13
 
14
  import os
15
  HF_TOKEN = os.environ["HF_TOKEN"]