From 3d477870adec2ee82228eb94251cc73edf4e67db Mon Sep 17 00:00:00 2001 From: "l.gabrysiak" Date: Sat, 1 Mar 2025 09:47:44 +0100 Subject: [PATCH] mod herbert --- herbert.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/herbert.py b/herbert.py index 153ed9c..675ddc3 100644 --- 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)