How to use sidmanale643/medLLAMA_3x with PEFT:
from peft import PeftModel from transformers import AutoModelForCausalLM base_model = AutoModelForCausalLM.from_pretrained("TinyPixel/Llama-2-7B-bf16-sharded") model = PeftModel.from_pretrained(base_model, "sidmanale643/medLLAMA_3x")
How to use sidmanale643/medLLAMA_3x with Transformers:
# Use a pipeline as a high-level helper # Warning: Pipeline type "question-answering" is no longer supported in transformers v5. # You must load the model directly (see below) or downgrade to v4.x with: # pip install "transformers<5.0.0" from transformers import pipeline pipe = pipeline("question-answering", model="sidmanale643/medLLAMA_3x")
# pip install -U transformers accelerate # Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("sidmanale643/medLLAMA_3x", device_map="auto")
· Sign up or log in to comment
