add automodel
Browse files- config.json +5 -1
config.json
CHANGED
@@ -15,5 +15,9 @@
|
|
15 |
"transformers_version": "4.42.4",
|
16 |
"triplane_dim": 80,
|
17 |
"triplane_high_res": 64,
|
18 |
-
"triplane_low_res": 32
|
|
|
|
|
|
|
|
|
19 |
}
|
|
|
15 |
"transformers_version": "4.42.4",
|
16 |
"triplane_dim": 80,
|
17 |
"triplane_high_res": 64,
|
18 |
+
"triplane_low_res": 32,
|
19 |
+
"auto_map": {
|
20 |
+
"AutoConfig": "modeling.LRMGeneratorConfig",
|
21 |
+
"AutoModel": "modeling.LRMGenerator"
|
22 |
+
}
|
23 |
}
|