jordigonzm commited on
Commit
45bb7b2
1 Parent(s): c86f71e

trust_remote_code=True

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,8 +8,8 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
8
 
9
  model_name = "microsoft/Phi-3-mini-128k-instruct"
10
 
11
- tokenizer = AutoTokenizer.from_pretrained(model_name)
12
- model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.bfloat16)
13
 
14
  pipeline = transformers.pipeline(
15
  "text-generation",
 
8
 
9
  model_name = "microsoft/Phi-3-mini-128k-instruct"
10
 
11
+ tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
12
+ model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.bfloat16, trust_remote_code=True)
13
 
14
  pipeline = transformers.pipeline(
15
  "text-generation",