Adding matplotlib to requirements. Removing import from train_utils

requirements.txt

@@ -12,4 +12,5 @@ transformers>=4.31.0
 sentencepiece
 py7zr
 scipy
-optimum
+optimum
+matplotlib

src/llama_recipes/utils/train_utils.py

@@ -16,7 +16,6 @@ from torch.distributed.fsdp import StateDictType
 from torch.distributed.fsdp.sharded_grad_scaler import ShardedGradScaler
 from tqdm import tqdm
 from transformers import LlamaTokenizer
-import matplotlib.pyplot as plt
 import json