Yntec commited on
Commit
a67522c
1 Parent(s): 411a64b

Update text_encoder/config.json

Browse files
Files changed (1) hide show
  1. text_encoder/config.json +1 -1
text_encoder/config.json CHANGED
@@ -18,7 +18,7 @@
18
  "num_hidden_layers": 12,
19
  "pad_token_id": 1,
20
  "projection_dim": 768,
21
- "torch_dtype": "float16",
22
  "transformers_version": "4.35.2",
23
  "vocab_size": 49408
24
  }
 
18
  "num_hidden_layers": 12,
19
  "pad_token_id": 1,
20
  "projection_dim": 768,
21
+ "torch_dtype": "float32",
22
  "transformers_version": "4.35.2",
23
  "vocab_size": 49408
24
  }