diff --git a/model.py b/model.py index 781251f..26c382d 100644 --- a/model.py +++ b/model.py @@ -3,7 +3,7 @@ from unsloth.chat_templates import get_chat_template from transformers import TextStreamer model, tokenizer = FastLanguageModel.from_pretrained( - model_name = "scoliono/groupchat_lora_abliterated_8b", + model_name = "scoliono/groupchat_lora_abliterated_instruct-3.1-8b", max_seq_length = 2048, dtype = None, load_in_4bit = True,