Update chat_jirack_1b_pt_packed.py

chat_jirack_1b_pt_packed.py

@@ -82,7 +82,9 @@ def main():
     device = torch.device("cuda")
     checkpoint = "jirack_1b_packed.pt"
     
-    tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-3.2-1B-Instruct")
+    #tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-3.2-1B-Instruct")
+    tokenizer = AutoTokenizer.from_pretrained("CMSManhattan/JiRackTokenizer")
+    
     
     # Создаем модель-наследника
     model = TernaryTransformer1B(TernaryConfig()).to(device=device, dtype=torch.float16)