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 <sayakpaul@users.noreply.huggingface.co>
- 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,
|