jfelipenc commited on
Commit
f198905
·
1 Parent(s): 030a756

Update app.py

Browse files

Fixing model_id parameter

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -69,7 +69,7 @@ tokenizer.padding_side = 'left'
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")
73
  peft_model = PeftModel.from_pretrained(peft_model, "Tonic/GaiaMiniMed")
74
 
75
 
 
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