Spaces:

Akiane-Cloudy
/

iOS_Final_Project

Build error

Akiane-Cloudy commited on 28 days ago

Commit

5808f7c

verified ·

1 Parent(s): aa7d0ae

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,7 +3,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 from transformers.generation import GenerationConfig
 import gradio as gr
-MODEL_NAME = "X-D-Lab/MindChat-Internlm2-7B"
 tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, trust_remote_code=True)
 if tokenizer.pad_token is None:
     tokenizer.pad_token = tokenizer.eos_token

 from transformers.generation import GenerationConfig
 import gradio as gr
+MODEL_NAME = "X-D-Lab/MindChat-Qwen-7B-v2"
 tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, trust_remote_code=True)
 if tokenizer.pad_token is None:
     tokenizer.pad_token = tokenizer.eos_token