Tonic commited on
Commit
47ca87a
1 Parent(s): 0caf514

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -61,9 +61,8 @@ tokenizer.padding_side = 'left'
61
 
62
  # Load the GaiaMiniMed model with the specified configuration
63
  # Load the Peft model with a specific configuration
64
- peft_config = PeftConfig.from_pretrained("Tonic/GaiaMiniMed")
65
  peft_model = PeftModel.from_pretrained("Tonic/GaiaMiniMed", config=peft_config)
66
-
67
  # Now you can use peft_model without any NameError
68
  peft_model = peft_model.to_bettertransformer("tiiuae/falcon-7b-instruct")
69
 
 
61
 
62
  # Load the GaiaMiniMed model with the specified configuration
63
  # Load the Peft model with a specific configuration
 
64
  peft_model = PeftModel.from_pretrained("Tonic/GaiaMiniMed", config=peft_config)
65
+ peft_model = AutoModelForCausalLM.from_pretrained("tiiuae/falcon-7b-instruct")
66
  # Now you can use peft_model without any NameError
67
  peft_model = peft_model.to_bettertransformer("tiiuae/falcon-7b-instruct")
68