add nvidia-cosmos-predict-2

Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2025-06-26 12:21:35 -04:00
parent 61e949c1cc
commit ada38d57b4
13 changed files with 150 additions and 7 deletions
+1
View File
@@ -42,6 +42,7 @@ pipelines = {
'Amused': getattr(diffusers, 'AmusedPipeline', None),
'HiDream': getattr(diffusers, 'HiDreamImagePipeline', None),
'OmniGenPipeline': getattr(diffusers, 'OmniGenPipeline', None),
'Cosmos': getattr(diffusers, 'Cosmos2TextToImagePipeline', None),
# dynamically imported and redefined later
'Meissonic': getattr(diffusers, 'DiffusionPipeline', None), # dynamically redefined and loaded in sd_models.load_diffuser