Disty0 commited on
Commit
19a25b6
1 Parent(s): 56e7c4c

Update safety_checker/config.json

Browse files
Files changed (1) hide show
  1. safety_checker/config.json +1 -1
safety_checker/config.json CHANGED
@@ -14,7 +14,7 @@
14
  "model_type": "clip_text_model",
15
  "num_attention_heads": 12
16
  },
17
- "torch_dtype": "float32",
18
  "transformers_version": "4.34.1",
19
  "vision_config": {
20
  "dropout": 0.0,
 
14
  "model_type": "clip_text_model",
15
  "num_attention_heads": 12
16
  },
17
+ "torch_dtype": "float16",
18
  "transformers_version": "4.34.1",
19
  "vision_config": {
20
  "dropout": 0.0,