fix error from prepare call running twice in a row

modules/training.py

@@ -90,6 +90,7 @@ def do_train(loraName: str, microBatchSize: int, batchSize: int, epochs: int, le
         evalData = load_dataset("json", data_files=cleanPath('training/datasets', f'{evalDataset}.json'))
         evalData = evalData['train'].shuffle().map(generate_and_tokenize_prompt)
     # Start prepping the model itself
+    if not hasattr(model, 'lm_head') or hasattr(model.lm_head, 'weight'):
         model = prepare_model_for_int8_training(model)
     config = LoraConfig(
         r=loraRank,