griptape_nodes 0.78.2__tar.gz → 0.78.3__tar.gz

{griptape_nodes-0.78.2 → griptape_nodes-0.78.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: griptape-nodes
-Version: 0.78.2
+Version: 0.78.3
 Summary: Add your description here
 Requires-Dist: griptape>=1.9.4
 Requires-Dist: pydantic>=2.10.6

{griptape_nodes-0.78.2 → griptape_nodes-0.78.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "griptape-nodes"
-version = "0.78.2"
+version = "0.78.3"
 description = "Add your description here"
 readme = "README.md"
 requires-python = ">=3.12.0, <3.13"

{griptape_nodes-0.78.2 → griptape_nodes-0.78.3}/src/griptape_nodes/exe_types/param_components/huggingface/huggingface_model_parameter.py

@@ -4,9 +4,6 @@ from abc import ABC, abstractmethod
 
 from griptape_nodes.exe_types.core_types import Parameter, ParameterMessage, ParameterMode
 from griptape_nodes.exe_types.node_types import BaseNode
-from griptape_nodes.exe_types.param_components.huggingface.huggingface_utils import (
-    assert_model_has_checkpoint_files,
-)
 from griptape_nodes.traits.options import Options
 
 logger = logging.getLogger("griptape_nodes")
@@ -120,15 +117,10 @@ class HuggingFaceModelParameter(ABC):
     def validate_before_node_run(self) -> list[Exception] | None:
         self.refresh_parameters()
         try:
-            repo_id, revision = self.get_repo_revision()
+            self.get_repo_revision()
         except Exception as e:
             return [e]
 
-        try:
-            assert_model_has_checkpoint_files(repo_id, revision)
-        except ValueError as e:
-            return [e]
-
         return None
 
     def list_repo_revisions(self) -> list[tuple[str, str]]:

{griptape_nodes-0.78.2 → griptape_nodes-0.78.3}/src/griptape_nodes/exe_types/param_components/huggingface/huggingface_utils.py

@@ -2,13 +2,11 @@ import logging
 from pathlib import Path
 from typing import Any
 
-from huggingface_hub import scan_cache_dir, snapshot_download
+from huggingface_hub import scan_cache_dir
 from huggingface_hub.constants import HF_HUB_CACHE
 
 logger = logging.getLogger("griptape_nodes")
 
-_WEIGHT_FILE_EXTENSIONS = (".bin", ".safetensors")
-
 
 def list_all_repo_revisions_in_cache() -> list[tuple[str, str]]:
     """Returns a list of (repo_id, revision) tuples for all repos in the huggingface cache."""
@@ -77,38 +75,6 @@ def list_repo_revisions_with_file_in_cache(repo_id: str, file: str) -> list[tupl
     return results
 
 
-def assert_model_has_checkpoint_files(repo_id: str, revision: str) -> None:
-    """Raise ValueError if a cached model's snapshot is missing or has no weight files.
-
-    Prevents cryptic Transformers/Diffusers errors when a model download is incomplete
-    or corrupt. Raises if the snapshot path cannot be resolved or if no weight files
-    (.bin, .safetensors) are present.
-
-    Args:
-        repo_id: HuggingFace repository ID (e.g. 'google/t5-v1_1-xxl')
-        revision: Model revision commit hash
-    """
-    try:
-        snapshot_path = Path(snapshot_download(repo_id=repo_id, revision=revision, local_files_only=True))
-    except Exception as e:
-        msg = (
-            f"Model '{repo_id}' appears to be corrupt or incomplete — "
-            "could not locate its snapshot in the model cache. "
-            "Please delete and re-download the model in the Model Manager."
-        )
-        raise ValueError(msg) from e
-
-    checkpoint_files = [f for ext in _WEIGHT_FILE_EXTENSIONS for f in snapshot_path.glob(f"*{ext}")]
-    if not checkpoint_files:
-        ext_list = ", ".join(_WEIGHT_FILE_EXTENSIONS)
-        msg = (
-            f"Model '{repo_id}' appears to be corrupt or incomplete — "
-            f"no checkpoint files ({ext_list}) found in the model cache. "
-            "Please delete and re-download the model in the Model Manager."
-        )
-        raise ValueError(msg)
-
-
 def quick_scan_diffuser_repos(cache_dir: str) -> list[dict[str, Any]]:
     """Quick scan of diffuser repositories in huggingface cache directory.