Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -6,7 +6,7 @@ peft_model_id = "JuliaUpton/Math_AI"
|
|
6 |
config = PeftConfig.from_pretrained(peft_model_id)
|
7 |
model = AutoModelForCausalLM.from_pretrained(config.base_model_name_or_path, return_dict=True, load_in_8bit=False)
|
8 |
tokenizer = AutoTokenizer.from_pretrained(config.base_model_name_or_path)
|
9 |
-
|
10 |
# Load the Lora model
|
11 |
merged_model = PeftModel.from_pretrained(model, peft_model_id)
|
12 |
|
|
|
6 |
config = PeftConfig.from_pretrained(peft_model_id)
|
7 |
model = AutoModelForCausalLM.from_pretrained(config.base_model_name_or_path, return_dict=True, load_in_8bit=False)
|
8 |
tokenizer = AutoTokenizer.from_pretrained(config.base_model_name_or_path)
|
9 |
+
|
10 |
# Load the Lora model
|
11 |
merged_model = PeftModel.from_pretrained(model, peft_model_id)
|
12 |
|