riffusion-model-v1 / safety_checker /model_config.json
jm12138's picture
add model
4bfd20a
raw
history blame contribute delete
No virus
368 Bytes
{
"init_args": [
{
"image_resolution": 224,
"vision_layers": 24,
"vision_heads": 16,
"vision_embed_dim": 1024,
"vision_patch_size": 14,
"vision_mlp_ratio": 4,
"vision_hidden_act": "quick_gelu",
"projection_dim": 768,
"init_class": "CLIPVisionModel"
}
],
"init_class": "StableDiffusionSafetyChecker"
}