Spaces:
Sleeping
Sleeping
v-yihangzhai
commited on
Commit
·
75ecf22
1
Parent(s):
e7167c7
require
Browse files
app.py
CHANGED
@@ -2,8 +2,8 @@ import gradio as gr
|
|
2 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
3 |
|
4 |
def vicuna(input):
|
5 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
6 |
-
model = AutoModelForCausalLM.from_pretrained("
|
7 |
model.half().cuda()
|
8 |
|
9 |
# prompt = """\
|
|
|
2 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
3 |
|
4 |
def vicuna(input):
|
5 |
+
tokenizer = AutoTokenizer.from_pretrained("CarperAI/stable-vicuna-13b-delta")
|
6 |
+
model = AutoModelForCausalLM.from_pretrained("CarperAI/stable-vicuna-13b-delta")
|
7 |
model.half().cuda()
|
8 |
|
9 |
# prompt = """\
|