update

model/models/__init__.py

@@ -1,5 +1,6 @@
 from .imagenhub_models import load_imagenhub_model
 from .playground_api import load_playground_model
+from .fal_api_models import load_fal_model
 
 IMAGE_GENERATION_MODELS = ['imagenhub_LCM_generation','imagenhub_SDXLTurbo_generation','imagenhub_SDXL_generation', 'imagenhub_PixArtAlpha_generation',
                             'imagenhub_OpenJourney_generation','imagenhub_SDXLLightning_generation', 'imagenhub_StableCascade_generation',
@@ -22,6 +23,8 @@ def load_pipeline(model_name):
         pipe = load_imagenhub_model(model_name, model_type)
     elif model_source == "playground":
         pipe = load_playground_model(model_name)
+    elif model_source == "fal":
+        raise NotImplementedError("FAL models are not supported yet")
     else:
         raise ValueError(f"Model source {model_source} not supported")
     return pipe

model/models/fal_api_models.py

@@ -42,6 +42,8 @@ class FalModel():
 
             result = handler.get()
             return result
+        elif self.model_type == "text2video":
+            raise NotImplementedError("text2video model is not implemented yet")
         else:
             raise ValueError("model_type must be text2image or image2image")