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)
|