valhalla's picture
add model
41c77df
{
"_class_name": "UNetLDMModel",
"_diffusers_version": "0.0.1",
"attention_resolutions": [
4,
2,
1
],
"channel_mult": [
1,
2,
4,
4
],
"context_dim": 1280,
"conv_resample": true,
"dims": 2,
"dropout": 0,
"image_size": 32,
"in_channels": 4,
"legacy": false,
"model_channels": 320,
"n_embed": null,
"name_or_path": "../fusing-models/unet/",
"num_classes": null,
"num_head_channels": -1,
"num_heads": 8,
"num_heads_upsample": -1,
"num_res_blocks": 2,
"out_channels": 4,
"resblock_updown": false,
"transformer_depth": 1,
"use_checkpoint": false,
"use_fp16": false,
"use_new_attention_order": false,
"use_scale_shift_norm": false,
"use_spatial_transformer": true
}