Add final model and weightsé

histaug_model.py

@@ -92,8 +92,8 @@ class Block(nn.Module):
         dim: int,
         num_heads: int,
         mlp_ratio: float = 4.0,
-        qkv_bias: bool = True,
-        qk_norm: bool = True,
+        qkv_bias: bool = False,
+        qk_norm: bool = False,
         proj_drop: float = 0.0,
         attn_drop: float = 0.0,
         init_values: float = None,

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:eb0b389f5ec2305de6e4cb1b39caf6b9ed204d8e0a9e9691b8d3e39dd3f9f009
-size 20617360
+oid sha256:f08b49be100d321670e4cd67afcbf1950bd164bd01d3e27d7c09ced5fd63cf9c
+size 20588864