WinterGYC
commited on
Commit
•
3ce93d7
1
Parent(s):
972ba8e
Fix import
Browse files
app.py
CHANGED
@@ -1,7 +1,6 @@
|
|
1 |
from transformers import AutoModel, AutoTokenizer
|
2 |
import gradio as gr
|
3 |
import mdtex2html
|
4 |
-
from utils import load_model_on_gpus
|
5 |
|
6 |
model = AutoModelForCausalLM.from_pretrained(
|
7 |
"baichuan-inc/Baichuan-13B-Chat",
|
|
|
1 |
from transformers import AutoModel, AutoTokenizer
|
2 |
import gradio as gr
|
3 |
import mdtex2html
|
|
|
4 |
|
5 |
model = AutoModelForCausalLM.from_pretrained(
|
6 |
"baichuan-inc/Baichuan-13B-Chat",
|