simonzhang5429 commited on
Commit
7f2ecf4
·
verified ·
1 Parent(s): f75a8b9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -2,7 +2,7 @@ import gradio as gr
2
  from transformers import AutoModelForCausalLM, AutoTokenizer
3
 
4
  TOKENIZER_REPO = "MediaTek-Research/Breeze-7B-Instruct-v1_0"
5
- tokenizer = AutoTokenizer.from_pretrained(TOKENIZER_REPO,local_files_only=True,use_fast=True,force_download=True)
6
  model = AutoModelForCausalLM.from_pretrained(
7
  TOKENIZER_REPO,
8
  device_map="auto",
 
2
  from transformers import AutoModelForCausalLM, AutoTokenizer
3
 
4
  TOKENIZER_REPO = "MediaTek-Research/Breeze-7B-Instruct-v1_0"
5
+ tokenizer = AutoTokenizer.from_pretrained(TOKENIZER_REPO,local_files_only=False,use_fast=True,force_download=True)
6
  model = AutoModelForCausalLM.from_pretrained(
7
  TOKENIZER_REPO,
8
  device_map="auto",