JustScriptzz commited on
Commit
a0d6d17
·
verified ·
1 Parent(s): d7e7476

Upload configuration_nexus_small.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. configuration_nexus_small.py +4 -1
configuration_nexus_small.py CHANGED
@@ -1,5 +1,5 @@
1
  from dataclasses import dataclass
2
- from transformers import PretrainedConfig
3
 
4
 
5
  @dataclass
@@ -65,3 +65,6 @@ class NexusSmAllConfig(PretrainedConfig):
65
  self.ff_dim = ff_dim
66
  self.norm_eps = norm_eps
67
  self.rope_theta = rope_theta
 
 
 
 
1
  from dataclasses import dataclass
2
+ from transformers import PretrainedConfig, AutoConfig
3
 
4
 
5
  @dataclass
 
65
  self.ff_dim = ff_dim
66
  self.norm_eps = norm_eps
67
  self.rope_theta = rope_theta
68
+
69
+
70
+ AutoConfig.register("nexus_small", NexusSmAllConfig)