diff --git a/herbert.py b/herbert.py index 675ddc3..3242cae 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 = "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)