Update elm/utils.py

elm/utils.py

@@ -1,21 +1,15 @@
 # Copyright (c) 2024, SliceX AI, Inc. All Rights Reserved.
 
-from prettytable import PrettyTable
-
 def count_parameters(model):
     """Count the number of parameters in the model."""
-    table = PrettyTable(["Modules", "Parameters"])
     total_params = 0
 
     for name, parameter in model.named_parameters():
         if not parameter.requires_grad: continue
         params = parameter.numel()
-        table.add_row([name, params])
         total_params+=params
     
-    print(table)
     print(f"Total Trainable Params: {total_params}")
-    
     return total_params