mvdream-hf / unet /config.json
Koke_Cacao
:bug: fix direct load from pipeline, thanks @reynoldscem
c4ac4f3
raw
history blame contribute delete
465 Bytes
{
"_class_name": "MultiViewUNetWrapperModel",
"_diffusers_version": "0.21.4",
"attention_resolutions": [
4,
2,
1
],
"camera_dim": 16,
"channel_mult": [
1,
2,
4,
4
],
"context_dim": 768,
"image_size": 32,
"in_channels": 4,
"legacy": false,
"model_channels": 320,
"num_heads": 8,
"num_res_blocks": 2,
"out_channels": 4,
"transformer_depth": 1,
"use_checkpoint": false,
"use_spatial_transformer": true
}