research14 commited on
Commit
dedc54d
1 Parent(s): 8eb0f9a
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,8 +3,8 @@ from transformers import GPT2LMHeadModel, GPT2Tokenizer
3
 
4
  # Load the Vicuna 7B v1.3 LMSys model and tokenizer
5
  model_name = "lmsys/vicuna-7b-v1.3"
6
- model = GPT2LMHeadModel.from_pretrained(model_name)
7
  tokenizer = GPT2Tokenizer.from_pretrained(model_name)
 
8
 
9
  with gr.Blocks() as demo:
10
  chatbot = gr.Chatbot()
 
3
 
4
  # Load the Vicuna 7B v1.3 LMSys model and tokenizer
5
  model_name = "lmsys/vicuna-7b-v1.3"
 
6
  tokenizer = GPT2Tokenizer.from_pretrained(model_name)
7
+ model = GPT2LMHeadModel.from_pretrained(model_name, force_download=True)
8
 
9
  with gr.Blocks() as demo:
10
  chatbot = gr.Chatbot()