Update fine_tuning.py

fine_tuning.py

@@ -159,6 +159,9 @@ def load_and_train(model_id="TinyLlama/TinyLlama-1.1B-Chat-v1.0"):
         data_collator=data_collator
     )
 
+    print(torch.cuda.is_available())  # True if GPU is detected
+    print(next(model.parameters()).device)  # Shows where your model is
+
     print("Training started")
     trainer.train()
     model.eval()