mod herbert

2025-03-01 09:47:44 +01:00 · 2025-03-01 09:47:44 +01:00 · 3d477870ad
parent a5e5401548
commit 3d477870ad
1 changed files with 1 additions and 1 deletions
--- a/herbert.py
+++ b/herbert.py
@ -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 = "pczarnik/herbert-base-ner"
+model_name = "Voicelab/trurl-2-13b"
 model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float16).to(device)
 tokenizer = AutoTokenizer.from_pretrained(model_name)