AminFaraji commited on
Commit
f6d009c
·
verified ·
1 Parent(s): d6f366b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -0
app.py CHANGED
@@ -41,6 +41,12 @@ from transformers import (
41
  )
42
 
43
 
 
 
 
 
 
 
44
  MODEL_NAME = "tiiuae/falcon-7b-instruct"
45
 
46
  model = AutoModelForCausalLM.from_pretrained(
 
41
  )
42
 
43
 
44
+ import subprocess
45
+ import sys
46
+
47
+ def install(package):
48
+ subprocess.check_call([sys.executable, "-m", "pip", "install", package])
49
+ install('accelerate')
50
  MODEL_NAME = "tiiuae/falcon-7b-instruct"
51
 
52
  model = AutoModelForCausalLM.from_pretrained(