Nvidia-CMU25
/

AutoregressiveFutureWorld

Model card Files Files and versions Community

sheldonl commited on Feb 3

Commit

eccdd96

·

1 Parent(s): 0962cc9

Fix config type problem

Files changed (1) hide show

inference_utils.py +1 -1

inference_utils.py CHANGED Viewed

@@ -281,7 +281,7 @@ def load_model_by_config(
     # config = importlib.import_module(config_module).make_config()
     if model_class in (DiffusionT2WModel, DiffusionV2WModel):
         config = make_config()
-    elif model_class in (LatentDiffusionDecoderModel):
         config = ldd_make_config()

     # config = importlib.import_module(config_module).make_config()
     if model_class in (DiffusionT2WModel, DiffusionV2WModel):
         config = make_config()
+    elif model_class is LatentDiffusionDecoderModel:
         config = ldd_make_config()