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

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)
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=True,use_fast=True,force_download=True)
6
  model = AutoModelForCausalLM.from_pretrained(
7
  TOKENIZER_REPO,
8
  device_map="auto",