Fal7acy
/

big-brain-lm

Text Generation

feature-extraction

Model card Files Files and versions Community

Fal7acy commited on Dec 11, 2023

Commit

fc99f44

•

1 Parent(s): 5cfa59c

Upload model

Files changed (2) hide show

language.py +1 -1
pytorch_model.bin +1 -1

language.py CHANGED Viewed

@@ -134,7 +134,7 @@ class BigBrainLanguageModel(PreTrainedModel):
     config_class = BigBrainConfig
     base_model_prefix = 'big-brain-lm'
-    def __init__(self, config: BigBrainConfig):
         super().__init__(config)
         self.config = config
         self.tok_embed = nn.Embedding(config.vocab_size, config.hidden_size, config.pad_token_id)

     config_class = BigBrainConfig
     base_model_prefix = 'big-brain-lm'
+    def __init__(self, config: BigBrainConfig = BigBrainConfig()):
         super().__init__(config)
         self.config = config
         self.tok_embed = nn.Embedding(config.vocab_size, config.hidden_size, config.pad_token_id)

pytorch_model.bin CHANGED Viewed

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c547697b1f4cc69295875fd3afc1679df421333ef53026d3f1f50cbc6b1dd5a3
 size 774713018

 version https://git-lfs.github.com/spec/v1
+oid sha256:4bc2647de306177bc7c1e1266bb7a6a98f8aee907e18e2b8d0d4457f4c1806d1
 size 774713018