diff --git a/.github/workflows/build-and-test.yml b/.github/workflows/build-and-test.yml
index 290df9b..e9a45cc 100644
--- a/.github/workflows/build-and-test.yml
+++ b/.github/workflows/build-and-test.yml
@@ -38,4 +38,8 @@ jobs:
           comfy --skip-prompt --no-enable-telemetry env
           comfy --skip-prompt install --cpu
           comfy launch --background -- --cpu
+          comfy model download --url https://huggingface.co/guoyww/animatediff/resolve/cd71ae134a27ec6008b968d6419952b0c0494cf2/mm_sd_v14.ckpt?download=true --relative-path models/animatediff_models --filename animatediff_models
+          comfy node install ComfyUI-AnimateDiff-Evolved
+          comfy node update ComfyUI-AnimateDiff-Evolved
+          comfy node uninstall ComfyUI-AnimateDiff-Evolved
           
\ No newline at end of file
diff --git a/comfy_cli/command/models/models.py b/comfy_cli/command/models/models.py
index e13ea70..ae5190a 100644
--- a/comfy_cli/command/models/models.py
+++ b/comfy_cli/command/models/models.py
@@ -20,7 +20,6 @@
 workspace_manager = WorkspaceManager()
 config_manager = ConfigManager()
 
-
 model_path_map = {
     "lora": "loras",
     "hypernetwork": "hypernetworks",
@@ -147,6 +146,13 @@ def download(
             show_default=True,
         ),
     ] = None,
+    filename: Annotated[
+        Optional[str],
+        typer.Option(
+            help="The filename to save the model.",
+            show_default=True,
+        ),
+    ] = None,
     set_civitai_api_token: Annotated[
         Optional[str],
         typer.Option(
@@ -230,9 +236,12 @@ def download(
     else:
         print("Model source is unknown")
 
-    local_filename = ui.prompt_input(
-        "Enter filename to save model as", default=local_filename
-    )
+    if filename is None:
+        local_filename = ui.prompt_input(
+            "Enter filename to save model as", default=local_filename
+        )
+    else:
+        local_filename = filename
 
     if relative_path is None:
         relative_path = DEFAULT_COMFY_MODEL_PATH
diff --git a/comfy_cli/registry/api.py b/comfy_cli/registry/api.py
index d77039a..5e68bef 100644
--- a/comfy_cli/registry/api.py
+++ b/comfy_cli/registry/api.py
@@ -18,6 +18,10 @@ def __init__(self):
     def determine_base_url(self):
         if os.getenv("ENVIRONMENT") == "dev":
             return "http://localhost:8080"
+        elif os.getenv("ENVIRONMENT") == "staging":
+            # print environment variable
+            print(os.getenv("ENVIRONMENT"))
+            return "https://staging-comfy-backend-qod3oz2v2q-uc.a.run.app"
         else:
             return "https://api.comfy.org"