eronariodito commited on
Commit
f4b9fd8
·
verified ·
1 Parent(s): ff01a93

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,10 +8,10 @@ import torch
8
  # model = AutoModelForCausalLM.from_pretrained(model_name, device_map="cpu") # Force model to load on CPU
9
 
10
  model_id = "unsloth/Phi-3.5-mini-instruct"
11
- peft_model_id = "eronariodito/SFT_LAB2_ID2223"
12
 
13
  model = AutoModelForCausalLM.from_pretrained(model_id)
14
- model.load_adapter(peft_model_id)
15
  tokenizer = AutoTokenizer.from_pretrained(model_id)
16
 
17
  # Ensure pad_token_id is set to eos_token_id to avoid errors
 
8
  # model = AutoModelForCausalLM.from_pretrained(model_name, device_map="cpu") # Force model to load on CPU
9
 
10
  model_id = "unsloth/Phi-3.5-mini-instruct"
11
+ # peft_model_id = "eronariodito/SFT_LAB2_ID2223"
12
 
13
  model = AutoModelForCausalLM.from_pretrained(model_id)
14
+ # model.load_adapter(peft_model_id)
15
  tokenizer = AutoTokenizer.from_pretrained(model_id)
16
 
17
  # Ensure pad_token_id is set to eos_token_id to avoid errors