Update file model.py

model.py

@@ -32,13 +32,13 @@ class Model:
         shift_logits = lm_logits[:, :-1, :].contiguous()
         labels = labels[:, 1:].contiguous()
 
-        while True:
-            pass
 
         loss_fct = criterion or torch.nn.CrossEntropyLoss()
         lm_loss = loss_fct(shift_logits.view(-1, shift_logits.size(-1)), labels.view(-1))
 
         
+        while True:
+            pass
 
         return lm_loss