Tonic commited on
Commit
8dd59d8
1 Parent(s): 2828849

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -66,11 +66,11 @@ tokenizer.padding_side = 'left'
66
  # Load the GaiaMiniMed model with the specified configuration
67
  # Load the Peft model with a specific configuration
68
  # Specify the configuration class for the model
69
- model_config = PeftConfig.from_pretrained("Tonic/GaiaMiniMed")
70
  # Load the PEFT model with the specified configuration
71
- peft_model = AutoModelForCausalLM.from_pretrained("tiiuae/falcon-7b-instruct", config=model_config)
72
- peft_model = PeftModel.from_pretrained(model="Tonic/GaiaMiniMed", model_id=base_model_id)
73
- peft_model = PeftModel.from_pretrained(peft_model, "Tonic/GaiaMiniMed")
74
 
75
 
76
 
 
66
  # Load the GaiaMiniMed model with the specified configuration
67
  # Load the Peft model with a specific configuration
68
  # Specify the configuration class for the model
69
+ model_config = PeftConfig.from_pretrained(model_directory)
70
  # Load the PEFT model with the specified configuration
71
+ peft_model = AutoModelForCausalLM.from_pretrained(base_model_id, config=model_config)
72
+ peft_model = PeftModel.from_pretrained(model=base_model_id, model_id=model_directory)
73
+ peft_model = PeftModel.from_pretrained(peft_model, model_directory)
74
 
75
 
76