eronariodito commited on
Commit
2e1a6a0
·
verified ·
1 Parent(s): f29169c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ 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(peft_model_id)
14
  # model.load_adapter(peft_model_id)
 
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/Llama_3.2_1B_SFT_The_Tome"
12
 
13
  model = AutoModelForCausalLM.from_pretrained(peft_model_id)
14
  # model.load_adapter(peft_model_id)