thuyentruong commited on
Commit
2ee3c08
·
verified ·
1 Parent(s): 6884b23

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -5
app.py CHANGED
@@ -8,15 +8,13 @@ import torch
8
 
9
  hf_token = os.environ.get('hf_token')
10
 
11
- model_path= 'microsoft/Phi-4-mini-instruct'
12
 
13
- model = FastLanguageModel.from_pretrained(
14
  model_path,
15
  #device_map="auto",
16
  #torch_dtype="auto",
17
- trust_remote_code=True,
18
- max_seq_length = max_seq_length,
19
- load_in_4bit = True,
20
  )
21
  tokenizer = AutoTokenizer.from_pretrained(model_path)
22
 
 
8
 
9
  hf_token = os.environ.get('hf_token')
10
 
11
+ model_path= 'unsloth/Phi-4-mini-instruct-bnb-4bit'#'microsoft/Phi-4-mini-instruct'
12
 
13
+ model = AutoModelForCausalLM.from_pretrained(
14
  model_path,
15
  #device_map="auto",
16
  #torch_dtype="auto",
17
+ trust_remote_code=True
 
 
18
  )
19
  tokenizer = AutoTokenizer.from_pretrained(model_path)
20