Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -69,7 +69,7 @@ tokenizer.padding_side = 'left'
|
|
69 |
model_config = AutoConfig.from_pretrained(base_model_id)
|
70 |
# Load the PEFT model with the specified configuration
|
71 |
peft_model = AutoModelForCausalLM.from_pretrained(model_directory, config=model_config)
|
72 |
-
peft_model = PeftModel.from_pretrained(model=base_model_id, trust_remote_code=True)
|
73 |
peft_model = PeftModel.from_pretrained(peft_model, model_directory)
|
74 |
|
75 |
|
|
|
69 |
model_config = AutoConfig.from_pretrained(base_model_id)
|
70 |
# Load the PEFT model with the specified configuration
|
71 |
peft_model = AutoModelForCausalLM.from_pretrained(model_directory, config=model_config)
|
72 |
+
peft_model = PeftModel.from_pretrained(model=base_model_id, model_id=model_directory, trust_remote_code=True)
|
73 |
peft_model = PeftModel.from_pretrained(peft_model, model_directory)
|
74 |
|
75 |
|