POKASH.PL
/
ably.do
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

mod herbert

This commit is contained in:
l.gabrysiak 2025-03-01 11:29:07 +01:00
parent 3d477870ad
commit 9bbe7188ca
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
herbert.py
View File

@ -54,7 +54,7 @@ eval_dataset = split_dataset["test"]
# 5️⃣ Ładowanie modelu Gemma 2B
device = "cuda" if torch.cuda.is_available() else "cpu"
model_name = "Voicelab/trurl-2-13b"
model_name = "mmosiolek/polpaca-lora-7b"
model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float16).to(device)
tokenizer = AutoTokenizer.from_pretrained(model_name)