Fix config type problem
Browse files- inference_utils.py +1 -1
inference_utils.py
CHANGED
|
@@ -281,7 +281,7 @@ def load_model_by_config(
|
|
| 281 |
# config = importlib.import_module(config_module).make_config()
|
| 282 |
if model_class in (DiffusionT2WModel, DiffusionV2WModel):
|
| 283 |
config = make_config()
|
| 284 |
-
elif model_class
|
| 285 |
config = ldd_make_config()
|
| 286 |
|
| 287 |
|
|
|
|
| 281 |
# config = importlib.import_module(config_module).make_config()
|
| 282 |
if model_class in (DiffusionT2WModel, DiffusionV2WModel):
|
| 283 |
config = make_config()
|
| 284 |
+
elif model_class is LatentDiffusionDecoderModel:
|
| 285 |
config = ldd_make_config()
|
| 286 |
|
| 287 |
|