Update config.json of UNet to include `attention_head_dim`
Browse files- 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,
|