namanbnsl
/

moose-mini

Model card Files Files and versions Community

namanbnsl commited on Jul 9

Commit

2135aec

•

1 Parent(s): ab74437

Update model.py

Files changed (1) hide show

model.py +1 -1

model.py CHANGED Viewed

@@ -403,4 +403,4 @@ class Moose(nn.Module):
         output = enc.decode(tokens.squeeze(0).tolist())
-        return output


403
404	output = enc.decode(tokens.squeeze(0).tolist())
405
406	+ return output[:-13]