diff --git a/modules/ltx/ltx_capabilities.py b/modules/ltx/ltx_capabilities.py index 78b56e89f..0d1909580 100644 --- a/modules/ltx/ltx_capabilities.py +++ b/modules/ltx/ltx_capabilities.py @@ -62,7 +62,10 @@ def _repo_cls_name(model_name: str) -> Optional[str]: if m.name == model_name: if m.repo_cls is None: return None - return m.repo_cls.__name__ + elif isinstance(m.repo_cls, str): + return m.repo_cls + else: + return m.repo_cls.__name__ return None diff --git a/modules/video_models/models_def.py b/modules/video_models/models_def.py index d0dd115c6..e2597fa0c 100644 --- a/modules/video_models/models_def.py +++ b/modules/video_models/models_def.py @@ -1,8 +1,6 @@ from dataclasses import dataclass import sys import time -import diffusers -import transformers from modules import errors from modules.logger import log @@ -656,13 +654,6 @@ try: for m in model: if m.name == 'None': continue - """ - if (m.repo_cls is None) or (m.dit_cls is None) or (m.te_cls is None): - log.error(f'Video: pipeline="{m.name}" not available') - errors += 1 - else: - total += 1 - """ total += 1 log.info(f'Networks: type="video" engines={len(models)} models={total} errors={errors} time={t1 - t0:.2f}') except Exception as e: