Views
No views yet
1from peft import PeftModel
2from transformers import AutoModelForCausalLM, AutoTokenizer
3
4base_model = AutoModelForCausalLM.from_pretrained("google/medgemma-1.5-4b-it")
5model = PeftModel.from_pretrained(base_model, "sahabajalam/Med_scribe")
6tokenizer = AutoTokenizer.from_pretrained("google/medgemma-1.5-4b-it")