Spaces:

smashmaster
/

tiny-scigpt-demo

Sleeping

Raymond commited on Sep 15

Commit

a7e1037

•

1 Parent(s): bb64cd3

fixed confusing comment

Files changed (1) hide show

model.py CHANGED Viewed

@@ -118,7 +118,7 @@ class Block(nn.Module):
         x = x + self.ffwd(self.ln2(x))
         return x
-# super simple bigram model
 class TokenBasedLanguageModel(nn.Module):
     def __init__(self):

         x = x + self.ffwd(self.ln2(x))
         return x
+# next token prediction model now
 class TokenBasedLanguageModel(nn.Module):
     def __init__(self):