Update model_index.json
Browse files- model_index.json +1 -0
model_index.json
CHANGED
|
@@ -1,6 +1,7 @@
|
|
| 1 |
{
|
| 2 |
"_version": 1,
|
| 3 |
"model": {
|
|
|
|
| 4 |
"type": "diffusion",
|
| 5 |
"architecture": "custom-diffusion",
|
| 6 |
"framework": "pytorch"
|
|
|
|
| 1 |
{
|
| 2 |
"_version": 1,
|
| 3 |
"model": {
|
| 4 |
+
"_class_name": "UNet2DConditionModel",
|
| 5 |
"type": "diffusion",
|
| 6 |
"architecture": "custom-diffusion",
|
| 7 |
"framework": "pytorch"
|