BeastGokul commited on
Commit
e4f30a4
·
verified ·
1 Parent(s): 6fe58d5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
4
  from peft import PeftModel, PeftConfig
5
 
6
  client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
7
- tokenizer = PeftConfig.from_pretrained("BeastGokul/Bio-Mistral-7B-finetuned")
8
  base_model = AutoModelForCausalLM.from_pretrained("BioMistral/BioMistral-7B")
9
  base_model.resize_token_embeddings(len(tokenizer))
10
  model = PeftModel.from_pretrained(base_model, "BeastGokul/Bio-Mistral-7B-finetuned")
 
4
  from peft import PeftModel, PeftConfig
5
 
6
  client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
7
+ tokenizer = AutoTokenizer.from_pretrained("BeastGokul/Bio-Mistral-7B-finetuned")
8
  base_model = AutoModelForCausalLM.from_pretrained("BioMistral/BioMistral-7B")
9
  base_model.resize_token_embeddings(len(tokenizer))
10
  model = PeftModel.from_pretrained(base_model, "BeastGokul/Bio-Mistral-7B-finetuned")