eronariodito commited on
Commit
81a8df6
·
verified ·
1 Parent(s): 4dbc6e6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,8 +7,8 @@ import torch
7
  # model_name = "PierreJousselin/lora_model" # Replace with the name you used on Hugging Face
8
  # model = AutoModelForCausalLM.from_pretrained(model_name, device_map="cpu") # Force model to load on CPU
9
 
10
- model_id = "unsloth/llama-3.2-3b-instruct-bnb-4bit"
11
- peft_model_id = "PierreJousselin/lora_model"
12
 
13
  model = AutoModelForCausalLM.from_pretrained(model_id)
14
  model.load_adapter(peft_model_id)
 
7
  # model_name = "PierreJousselin/lora_model" # Replace with the name you used on Hugging Face
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)