Hugging Face's logo

flopml
/
mamba

Model card Files
xet
Community
flpelerin commited on
Commit
8b5543e
·
1 Parent(s): 4c50abe

Update file model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -2
model.py CHANGED
@@ -28,8 +28,7 @@ class Model:
28
 
29
 
30
  def compute_loss(self, input_ids, labels=None, criterion=None):
31
- #lm_logits = self.model(input_ids).logits
32
- lm_logits = self.model(input_ids)
33
 
34
  labels = input_ids.to(GetDevice())
35
  shift_logits = lm_logits[:, :-1, :].contiguous()
 
28
 
29
 
30
  def compute_loss(self, input_ids, labels=None, criterion=None):
31
+ lm_logits = self.model(input_ids).logits
 
32
 
33
  labels = input_ids.to(GetDevice())
34
  shift_logits = lm_logits[:, :-1, :].contiguous()