Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2from peft import PeftModel
3
4base_model = "facebook/opt-350m" # <- BASE MODEL NAME HERE
5adapter_repo = "Amogh-2404/LegalGPT"
6
7model = AutoModelForCausalLM.from_pretrained(base_model)
8tokenizer = AutoTokenizer.from_pretrained(adapter_repo)
9model = PeftModel.from_pretrained(model, adapter_repo)1from transformers import AutoModelForCausalLM, AutoTokenizer
2from peft import PeftModel
3
4base_model = "facebook/opt-350m" # <- BASE MODEL NAME HERE
5adapter_repo = "Amogh-2404/LegalGPT"
6
7model = AutoModelForCausalLM.from_pretrained(base_model)
8tokenizer = AutoTokenizer.from_pretrained(adapter_repo)
9model = PeftModel.from_pretrained(model, adapter_repo)