Update model_index.json to use GPT2LMHeadModel rather than GPT2Model
#4
by
Mekadrom
- opened
- model_index.json +1 -1
model_index.json
CHANGED
|
@@ -7,7 +7,7 @@
|
|
| 7 |
],
|
| 8 |
"language_model": [
|
| 9 |
"transformers",
|
| 10 |
-
"
|
| 11 |
],
|
| 12 |
"projection_model": [
|
| 13 |
"audioldm2",
|
|
|
|
| 7 |
],
|
| 8 |
"language_model": [
|
| 9 |
"transformers",
|
| 10 |
+
"GPT2LMHeadModel"
|
| 11 |
],
|
| 12 |
"projection_model": [
|
| 13 |
"audioldm2",
|