Hugging Face's logo

flopml
/
mamba

Model card Files
xet
Community
flpelerin commited on
Commit
cb76b78
·
1 Parent(s): ca55198

Update file model.py

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