IMvision12 commited on
Commit
0c5039b
·
verified ·
1 Parent(s): 1bbb63c

Fix HUB-02: correct declared weights filename

Browse files
Files changed (1) hide show
  1. zm_config.json +2 -2
zm_config.json CHANGED
@@ -4,7 +4,7 @@
4
  "model_module": "zeromodels.models.gpt2",
5
  "model_class": "GPT2Model",
6
  "variant": "gpt2_large",
7
- "weights": "model.weights.json",
8
  "schema_version": 2,
9
  "weight_dtype": "float32",
10
  "model_type": "gpt2",
@@ -18,4 +18,4 @@
18
  "norm_eps": 1e-05,
19
  "tie_embeddings": true
20
  }
21
- }
 
4
  "model_module": "zeromodels.models.gpt2",
5
  "model_class": "GPT2Model",
6
  "variant": "gpt2_large",
7
+ "weights": "model.weights.h5",
8
  "schema_version": 2,
9
  "weight_dtype": "float32",
10
  "model_type": "gpt2",
 
18
  "norm_eps": 1e-05,
19
  "tie_embeddings": true
20
  }
21
+ }