Update file model.py

model.py

@@ -15,7 +15,7 @@ class Model:
     def Log(self):
         model_size, rounded_model_size = GetNumParams(self.model)
         print(f"Model has {model_size} ({rounded_model_size}) parameters")
-        print(f"Model's embedding size is {self.model.vocab_size}")
+        print(f"Model's embedding size is {self.vocab_size}")
 
 
     def AutoRegressiveLossFunction(self, input_ids, labels=None, criterion=None):