matthewyuan commited on
Commit
e11ef26
·
verified ·
1 Parent(s): 5d0c642

Upload config.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. config.json +6 -9
config.json CHANGED
@@ -1,11 +1,8 @@
1
  {
2
- "clip_embed_dim": 512,
3
- "clip_model": "ViT-B-32",
4
- "dropout": 0.3,
5
- "hidden_dim": 256,
6
- "model_type": "deep",
7
- "output_range": [
8
- 1.0,
9
- 10.0
10
- ]
11
  }
 
1
  {
2
+ "model_type": "iqf_model",
3
+ "class_name": "IQFModel",
4
+ "init_kwargs": {
5
+ "in_dim": 512,
6
+ "hidden": 256
7
+ }
 
 
 
8
  }