Tonic commited on
Commit
030a756
1 Parent(s): 7445a13

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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_id="Tonic/GaiaMiniMed")
73
  peft_model = PeftModel.from_pretrained(peft_model, "Tonic/GaiaMiniMed")
74
 
75
 
@@ -81,9 +81,9 @@ peft_model = PeftModel.from_pretrained(peft_model, "Tonic/GaiaMiniMed")
81
  #peft_model = AutoModelForCausalLM.from_pretrained(base_model_id, config=model_config)
82
 
83
  # Load the PEFT model
84
- # peft_config = PeftConfig.from_pretrained("Tonic/mistralmed", token="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")
85
  # peft_model = MistralForCausalLM.from_pretrained("mistralai/Mistral-7B-v0.1", trust_remote_code=True)
86
- # peft_model = PeftModel.from_pretrained(peft_model, "Tonic/mistralmed", token="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")
87
 
88
  class ChatBot:
89
  def __init__(self, system_prompt="You are an expert medical analyst:"):
 
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
 
 
81
  #peft_model = AutoModelForCausalLM.from_pretrained(base_model_id, config=model_config)
82
 
83
  # Load the PEFT model
84
+ # peft_config = PeftConfig.from_pretrained("Tonic/mistralmed")
85
  # peft_model = MistralForCausalLM.from_pretrained("mistralai/Mistral-7B-v0.1", trust_remote_code=True)
86
+ # peft_model = PeftModel.from_pretrained(peft_model, "Tonic/mistralmed")
87
 
88
  class ChatBot:
89
  def __init__(self, system_prompt="You are an expert medical analyst:"):