Update utils.py

utils.py

@@ -92,7 +92,6 @@ def load_tokenizer_and_model_Blaize(base_model, load_8bit=True):
     
     tokenizer = LlamaTokenizer.from_pretrained(base_model, add_eos_token=True, use_auth_token=True)
     model = LlamaForCausalLM.from_pretrained(base_model, load_in_8bit=True, device_map="auto")
-    model = prepare_model_for_int8_training(model)
 
     return tokenizer,model, device