JuliaUpton commited on
Commit
a67e26b
1 Parent(s): 8d63e99

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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