Tonic commited on
Commit
8d2dad8
1 Parent(s): e5d09cd

Update app.py

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