FlawedLLM commited on
Commit
9df6fd0
·
verified ·
1 Parent(s): 4ca1aef

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -2
app.py CHANGED
@@ -1,3 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
  import os
2
  HF_TOKEN = os.environ["HF_TOKEN"]
3
  import re
@@ -5,8 +17,6 @@ import spaces
5
  import gradio as gr
6
  import torch
7
  from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig, AutoConfig
8
- from huggingface_hub import login, HfFolder
9
- pip list
10
  # from peft import PeftModel, PeftConfig
11
 
12
 
 
1
+
2
+ # from huggingface_hub import login, HfFolder
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"]
15
  import re
 
17
  import gradio as gr
18
  import torch
19
  from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig, AutoConfig
 
 
20
  # from peft import PeftModel, PeftConfig
21
 
22