emya commited on
Commit
8ad2f92
·
1 Parent(s): 54673f1

use 8bit model

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,7 +2,7 @@ import gradio as gr
2
 
3
  from transformers import pipeline
4
 
5
- pipe = pipeline("translation", model="emya/vicuna-7b-v1.5-steve-jobs-v1")
6
 
7
  def predict(text):
8
  prompt = f"{text} (Answer in a few sentences)"
@@ -14,4 +14,4 @@ demo = gr.Interface(
14
  outputs='text',
15
  )
16
 
17
- demo.launch()
 
2
 
3
  from transformers import pipeline
4
 
5
+ pipe = pipeline("translation", model="emya/vicuna-7b-v1.5-steve-jobs-8bit-v1")
6
 
7
  def predict(text):
8
  prompt = f"{text} (Answer in a few sentences)"
 
14
  outputs='text',
15
  )
16
 
17
+ demo.launch()