Commit
•
39e1979
1
Parent(s):
6c4e9e7
Update config.json of UNet to include `attention_head_dim` (#12)
Browse files- Update config.json of UNet to include `attention_head_dim` (f73e31411e1087b6dbddcea9611fbcfdec6e2efe)
Co-authored-by: Sayak Paul <[email protected]>
- 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,
|