Update config.json of UNet to include `attention_head_dim`
#12
by
sayakpaul
HF staff
- opened
- unet/config.json +1 -0
unet/config.json
CHANGED
@@ -2,6 +2,7 @@
|
|
2 |
"_class_name": "I2VGenXLUNet",
|
3 |
"_diffusers_version": "0.26.1",
|
4 |
"_name_or_path": "i2vgen-xl/unet",
|
|
|
5 |
"block_out_channels": [
|
6 |
320,
|
7 |
640,
|
|
|
2 |
"_class_name": "I2VGenXLUNet",
|
3 |
"_diffusers_version": "0.26.1",
|
4 |
"_name_or_path": "i2vgen-xl/unet",
|
5 |
+
"attention_head_dim": 64,
|
6 |
"block_out_channels": [
|
7 |
320,
|
8 |
640,
|