Update model_setup.py

model_setup.py

@@ -29,7 +29,7 @@ def load_model():
         device_map={"": "cpu"},  # Đặt sử dụng CPU
         token = token
     )
-    model.config.pretraining_tp = 1
+    # model.config.pretraining_tp = 1
 
     tokenizer = AutoTokenizer.from_pretrained('anhvv200053/Vinallama-2-7B-updated1-instruction-v2', trust_remote_code=True, use_fast=True, token = token)
     tokenizer.pad_token = tokenizer.eos_token