Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -62,7 +62,7 @@ tokenizer = AutoTokenizer.from_pretrained(base_model_id, trust_remote_code=True,
|
|
| 62 |
# Load the GaiaMiniMed model with the specified configuration
|
| 63 |
# Load the Peft model with a specific configuration
|
| 64 |
# Specify the configuration class for the model
|
| 65 |
-
model_config = AutoConfig.from_pretrained(
|
| 66 |
# Load the PEFT model with the specified configuration
|
| 67 |
peft_model = AutoModelForCausalLM.from_pretrained(model_directory, config=model_config)
|
| 68 |
peft_model = PeftModel.from_pretrained(peft_model, model_directory)
|
|
|
|
| 62 |
# Load the GaiaMiniMed model with the specified configuration
|
| 63 |
# Load the Peft model with a specific configuration
|
| 64 |
# Specify the configuration class for the model
|
| 65 |
+
model_config = AutoConfig.from_pretrained(base_model_id)
|
| 66 |
# Load the PEFT model with the specified configuration
|
| 67 |
peft_model = AutoModelForCausalLM.from_pretrained(model_directory, config=model_config)
|
| 68 |
peft_model = PeftModel.from_pretrained(peft_model, model_directory)
|