Views
No views yet
Note: This repository does not contain the base model weights. You must load the base model to use this adapter.
peft and transformers.1import torch
2from transformers import AutoModelForCausalLM, AutoTokenizer
3from peft import PeftModel
4
5# 1. Setup IDs
6base_model_id = "BSC-LT/salamandra-7b-instruct"
7adapter_id = "luisibear/salamandra-cot-lora"
8
9# 2. Load Tokenizer
10tokenizer = AutoTokenizer.from_pretrained(base_model_id)
11
12# 3. Load Base Model
13base_model = AutoModelForCausalLM.from_pretrained(
14 base_model_id,
15 torch_dtype=torch.float16,
16 device_map="auto"
17)
18
19# 4. Load Adapter
20model = PeftModel.from_pretrained(base_model, adapter_id)
21
22# 5. Inference Example
23messages = [
24 {"role": "user", "content": "Explain the logic behind the Pythagorean theorem step by step."}
25]
26
27# Apply chat template (ensure your tokenizer has the correct template set)
28inputs = tokenizer.apply_chat_template(messages, return_tensors="pt", add_generation_prompt=True).to("cuda")
29
30outputs = model.generate(inputs, max_new_tokens=512)
31print(tokenizer.decode(outputs[0], skip_special_tokens=True))```