Upload model

config.json

@@ -1,5 +1,5 @@
 {
-  "L_eff": 25,
+  "L_eff": 8,
   "architectures": [
     "NQSModel"
   ],

vit_fnqs_config.py

@@ -7,7 +7,7 @@ class ViTFNQSConfig(PretrainedConfig):
 
     def __init__(
         self,
-        L_eff=25,
+        L_eff=8,
         num_layers = 6,
         d_model = 72,
         heads = 12,

vit_fnqs_model.py

@@ -9,7 +9,7 @@ class ViTFNQSModel(FlaxPreTrainedModel):
     def __init__(
             self, 
             config: ViTFNQSConfig, 
-            input_shape = (jnp.zeros((1, 100)), jnp.zeros((1, 1))),
+            input_shape = (jnp.zeros((1, 32)), jnp.zeros((1, 32))),
             seed: int = 0,
             dtype: jnp.dtype = jnp.float64,
             _do_init: bool = True,