griptape-nodes 0.35.0__tar.gz → 0.36.0__tar.gz

griptape_nodes-0.36.0/.env.example

@@ -0,0 +1,2 @@
+GT_CLOUD_API_KEY=<your_api_key>
+GTN_USE_WEBSOCKETS=false

griptape_nodes-0.36.0/.github/workflows/claude.yml

@@ -0,0 +1,37 @@
+name: Claude Code
+
+on:
+  issue_comment:
+    types: [created]
+  pull_request_review_comment:
+    types: [created]
+  issues:
+    types: [opened, assigned]
+  pull_request_review:
+    types: [submitted]
+
+jobs:
+  claude:
+    if: |
+      (github.event_name == 'issue_comment' && contains(github.event.comment.body, '@claude')) ||
+      (github.event_name == 'pull_request_review_comment' && contains(github.event.comment.body, '@claude')) ||
+      (github.event_name == 'pull_request_review' && contains(github.event.review.body, '@claude')) ||
+      (github.event_name == 'issues' && (contains(github.event.issue.body, '@claude') || contains(github.event.issue.title, '@claude')))
+    runs-on: ubuntu-latest
+    permissions:
+      contents: read
+      pull-requests: read
+      issues: read
+      id-token: write
+    steps:
+      - name: Checkout repository
+        uses: actions/checkout@v4
+        with:
+          fetch-depth: 1
+
+      - name: Run Claude Code
+        id: claude
+        uses: anthropics/claude-code-action@beta
+        with:
+          anthropic_api_key: ${{ secrets.ANTHROPIC_API_KEY }}
+

griptape_nodes-0.36.0/.github/workflows/workflow-tests.yml

@@ -0,0 +1,52 @@
+name: Workflow Tests
+
+on:
+  push:
+    branches: [ "main" ]
+  pull_request:
+    branches: [ "main" ]
+  merge_group:
+    types: [checks_requested]
+
+concurrency:
+  group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref }}
+  cancel-in-progress: true
+
+jobs:
+  test-ubuntu:
+    runs-on: ubuntu-latest
+    env:
+      GT_CLOUD_API_KEY: ${{ secrets.GT_CLOUD_API_KEY }}
+    strategy:
+      fail-fast: false
+      matrix:
+        python-version: ["3.12"]
+    steps:
+      - name: Checkout actions
+        uses: actions/checkout@v4
+      - name: Init environment 
+        uses: ./.github/actions/init-environment 
+      - name: Run workflow tests
+        run: make test/workflows
+  test-windows:
+    runs-on: windows-latest
+    env:
+      GT_CLOUD_API_KEY: ${{ secrets.GT_CLOUD_API_KEY }}
+    strategy:
+      fail-fast: false
+      matrix:
+        python-version: ["3.12"]
+    defaults:
+      run:
+        shell: bash
+    steps:
+      - name: Checkout actions
+        uses: actions/checkout@v4
+      - name: Init environment 
+        uses: ./.github/actions/init-environment
+      - name: Run workflow tests
+        # TODO: make test/unit fails with the following error:
+        # process_begin: CreateProcess(NULL, uv run pytest -n auto tests/unit, ...) failed.
+        # make (e=2): The system cannot find the file specified.
+        # make: *** [Makefile:24: test/unit] Error 2
+        run: uv run pytest -s tests/workflows

{griptape_nodes-0.35.0 → griptape_nodes-0.36.0}/CONTRIBUTING.md

@@ -65,10 +65,10 @@ When developing, you typically want to run the engine using your local source co
     uv run gtn init
     ```
 
-- **Run Tests:**
+- **Run Unit Tests:**
 
     ```shell
-    uv run pytest
+    uv run pytest test/unit
     ```
 
     Or use the Makefile shortcut:
@@ -77,6 +77,9 @@ When developing, you typically want to run the engine using your local source co
     make test/unit
     ```
 
+    > Other test targets (e.g. `test/integration` and `test/workflow`) require
+    > a `.env` in the repo root with all keys found in `.env.example` to run.
+
 - **Check Code (Linting & Formatting):**
 
     ```shell

{griptape_nodes-0.35.0 → griptape_nodes-0.36.0}/Makefile

@@ -100,7 +100,7 @@ check/spell:
 	@uv run typos 
 
 .PHONY: test  ## Run all tests.
-test: test/unit test/integration
+test: test/unit test/integration test/workflows
 
 .PHONY: test/unit
 test/unit: ## Run unit tests.
@@ -112,6 +112,10 @@ test/integration: ## Run integration tests.
 	@uv run pytest -n auto tests/integration
 	@uv run pytest -n auto libraries/griptape_nodes_library/tests/integration
 
+.PHONY: test/workflows
+test/workflows: ## Run workflow tests.
+	@uv run pytest -s tests/workflows
+
 .PHONY: docs
 docs: ## Build documentation.
 	@uv run python -m mkdocs build --clean --strict 

{griptape_nodes-0.35.0 → griptape_nodes-0.36.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: griptape-nodes
-Version: 0.35.0
+Version: 0.36.0
 Summary: Add your description here
 License-File: LICENSE
 Requires-Python: ~=3.12

{griptape_nodes-0.35.0/docs/reference → griptape_nodes-0.36.0/docs}/command_line_interface.md

@@ -1,6 +1,6 @@
-# Griptape Nodes Command Line Documentation
+# Griptape Nodes Command Line Interface (CLI)
 
-This page describes what can be done in the terminal (or PowerShell) regarding the `griptape-nodes` (or `gtn`) command. For those who may be new to command-line interfaces (CLIs), a CLI is a text-based way to interact with software by typing commands instead of clicking buttons in a UI.
+If you're new to command-line interfaces (CLIs), a CLI is a text-based way to interact with software by typing commands instead of clicking buttons in a UI. Griptape Nodes provides a CLI with the the `griptape-nodes` (or `gtn`) command. This enables you to interact with Griptape Nodes in the terminal (and via PowerShell).
 
 `griptape-nodes` (or its shorthand alias `gtn`) is a command-line tool specifically designed to launch, update, and manage the Griptape Nodes Engine installation on your computer. This tool handles tasks like initializing your workspace, updating the software, managing configuration settings, and starting the engine that powers the web-based Griptape Nodes editor. The actual creation and editing of workflows happens in the web interface that opens when you run the engine.
 

{griptape_nodes-0.35.0 → griptape_nodes-0.36.0}/docs/ftue/00_tour/FTUE_00_tour.md

@@ -75,7 +75,7 @@ The most important area to focus on initially is the left panel, the node librar
 
 ## Adding Nodes to the Workspace
 
-There are three interactive methods to creating nodes (and even more in [Retained Mode](../../reference/retained_mode.md))
+There are three interactive methods to creating nodes (and even more in [Retained Mode](../../retained_mode.md))
 
 <div style="display: flex; justify-content: space-between; gap: 20px; margin-bottom: 30px;">
   <div style="flex: 1;">

{griptape_nodes-0.35.0 → griptape_nodes-0.36.0}/docs/ftue/FTUE.md

@@ -1,4 +1,4 @@
-# Lesson Overview
+# Tutorial Overview
 
 Welcome to your journey with Griptape Nodes! This learning path will guide you through everything from basic setup to creating complex, coordinated workflows with multiple agents.
 

{griptape_nodes-0.35.0/docs/reference → griptape_nodes-0.36.0/docs}/glossary.md

@@ -1,10 +1,10 @@
-# Griptape Nodes Glossary for Artists
+# Griptape Nodes Glossary for Artists & Creators
 
 ## What is Griptape Nodes?
 
-Griptape Nodes is a creative toolkit (built with Python) that helps artists and creators build AI-powered projects without needing deep technical expertise. Think of it like a set of building blocks that you can connect together to create art, generate images, process text, or even build other workhorse applications.
+Griptape Nodes is a toolkit that enables artists and creators to build AI-powered projects without the need for deep technical expertise. You can think of Griptape Nodes as a set of building blocks that you can connect together to create art, generate images, process text, or even build other workflow-centric applications.
 
-### The Big Ones in the Griptape Nodes world
+### Core Griptape Nodes Concepts
 
 - **Workflow**: A document containing nodes, connections, and values. While technically a Workflow is also a Script, avoid calling them Scripts, so that the term Script can communicate a clearly different thing than the term Workflow. This is also what we call the saved file.
 

{griptape_nodes-0.35.0 → griptape_nodes-0.36.0}/docs/index.md

@@ -12,5 +12,5 @@ With Griptape Nodes, you can connect visual building blocks that represent powe
 
 Whether you're a creative who wants to explore new AI-powered possibilities, a knowledge worker looking to automate tedious tasks, or a curious mind wanting to experiment with cutting-edge technology, Griptape Nodes opens doors you might have thought were locked to all but professional programmers.
 
-And for those who want to go deeper? A little Python knowledge goes a long way here. Our [simple scripting interface](reference/retained_mode.md) lets you customize and extend what's possible, turning your big ideas into working solutions without requiring an engineering degree.
+And for those who want to go deeper? A little Python knowledge goes a long way here. Our [simple scripting interface](retained_mode.md) lets you customize and extend what's possible, turning your big ideas into working solutions without requiring an engineering degree.
 Griptape Nodes is where your imagination meets AI capability. Let's build something amazing together.

{griptape_nodes-0.35.0 → griptape_nodes-0.36.0}/docs/installation.md

@@ -69,7 +69,7 @@ After typing and running `griptape-nodes` or `gtn` in the terminal *for the firs
 
 ## 3. Configuration
 
-**First**, you'll be prompted to set your *workspace directory*. Your workspace directory is where the Griptape Nodes engine will save [project files](./reference/glossary.md#project-files), and [generated assets](./reference/glossary.md#generated-assets). It will also contain a [.env](./reference/glossary.md#.env) for your Griptape Nodes [secret keys](./reference/glossary.md#secret-keys).
+**First**, you'll be prompted to set your *workspace directory*. Your workspace directory is where the Griptape Nodes engine will save [project files](./glossary.md#project-files), and [generated assets](./glossary.md#generated-assets). It will also contain a [.env](./glossary.md#.env) for your Griptape Nodes [secret keys](./glossary.md#secret-keys).
 
 ```
 ╭───────────────────────────────────────────────────────────────────╮

griptape_nodes-0.36.0/docs/nodes/config/groq_prompt_driver.md

@@ -0,0 +1,58 @@
+# Groq Prompt Driver
+
+The Groq Prompt Driver node allows you to configure and utilize Groq's language models within the Griptape Nodes framework. This node provides access to Groq's high-performance inference engine and various language models.
+
+## Configuration
+
+### API Key
+
+To use the Groq Prompt Driver, you need to provide a Groq API key. You can obtain an API key from the [Groq Console](https://console.groq.com/keys).
+
+### Available Models
+
+The Groq Prompt Driver supports the following models:
+
+#### Production Models
+
+- [gemma2-9b-it](https://huggingface.co/google/gemma-2-9b-it) - Google's Gemma 2 9B model
+- [meta-llama/llama-guard-4-12b](https://console.groq.com/docs/model/llama-guard-4-12b) - Meta's Llama Guard model for content moderation
+- [llama-3.3-70b-versatile](https://console.groq.com/docs/model/llama-3.3-70b-versatile) - Meta's versatile 70B parameter model
+- [llama-3.1-8b-instant](https://console.groq.com/docs/model/llama-3.1-8b-instant) - Meta's fast 8B parameter model
+- [llama3-70b-8192](https://console.groq.com/docs/model/llama3-70b-8192) - Meta's 70B parameter model with 8K context
+- [llama3-8b-8192](https://console.groq.com/docs/model/llama3-8b-8192) - Meta's 8B parameter model with 8K context
+- [meta-llama/llama-4-scout-17b-16e-instruct](https://console.groq.com/docs/model/llama-4-scout-17b-16e-instruct) - Meta's vision model with 16K context, compatible with the Image Description node
+- [meta-llama/llama-4-maverick-17b-128e-instruct](https://console.groq.com/docs/model/llama-4-maverick-17b-128e-instruct) - Meta's vision model with 128K context, compatible with the Image Description node
+
+#### Preview Models
+
+- [allam-2-7b](https://ai.azure.com/explore/models/ALLaM-2-7b-instruct/version/2/registry/azureml) - Saudi Data and AI Authority's 7B parameter model
+- [deepseek-r1-distill-llama-70b](https://console.groq.com/docs/model/deepseek-r1-distill-llama-70b) - DeepSeek's distilled 70B parameter model
+
+### Parameters
+
+The Groq Prompt Driver supports the following configuration parameters:
+
+| Parameter    | Type    | Default      | Description                                          |
+| ------------ | ------- | ------------ | ---------------------------------------------------- |
+| model        | string  | gemma2-9b-it | The Groq model to use for text generation            |
+| temperature  | float   | 0.7          | Controls randomness in the output (0.0 to 1.0)       |
+| top_p        | float   | 0.9          | Controls diversity via nucleus sampling (0.0 to 1.0) |
+| max_tokens   | integer | 2048         | Maximum number of tokens to generate                 |
+| stream       | boolean | false        | Whether to stream the response                       |
+| max_attempts | integer | 3            | Maximum number of retry attempts for failed requests |
+
+## Usage
+
+1. Add the Groq Prompt Driver node to your workflow
+1. Make sure to get an API key here: [Groq API Keys](https://console.groq.com/keys)
+1. Configure your `GROQ_API_KEY` in the Griptape Configuration settings
+1. Select your desired model from the available options
+1. Adjust the generation parameters as needed
+1. Connect the node to other nodes in your workflow
+
+## Notes
+
+- The Groq Prompt Driver uses the OpenAI-compatible API endpoint at `https://api.groq.com/openai/v1`
+- Preview models are intended for evaluation purposes and may be discontinued at short notice
+- Production models are recommended for production environments
+- The node automatically handles API key validation before workflow execution

{griptape_nodes-0.35.0/docs/reference → griptape_nodes-0.36.0/docs}/retained_mode.md

@@ -1,6 +1,6 @@
-# Griptape Nodes Scripting
+# Griptape Nodes Retained Mode Scripting
 
-"Retained Mode" is a Python scripting interface to interact with Griptape Nodes. These methods allow users to create, modify, and manage nodes, parameters, connections, and flows through a simplified Python API.
+"Retained Mode" provides a Python scripting interface for interacting with Griptape Nodes. This allows users to create, modify, and manage nodes, parameters, connections, and flows through a simplified Python API.
 
 > **Note:** The actual import command for RetainedMode is:
 >

{griptape_nodes-0.35.0 → griptape_nodes-0.36.0}/entrypoint.sh

@@ -2,7 +2,7 @@
 set -e
 
 if [ "$GTN_INIT" = "true" ]; then
-    /griptape-nodes/.venv/bin/griptape-nodes init
+    /griptape-nodes/.venv/bin/griptape-nodes init --no-interactive
 fi
 
 exec "$@"

{griptape_nodes-0.35.0 → griptape_nodes-0.36.0}/libraries/griptape_nodes_advanced_media_library/diffusers_nodes_library/common/parameters/huggingface_model_parameter.py

@@ -2,7 +2,7 @@ import logging
 from abc import ABC, abstractmethod
 
 from diffusers_nodes_library.common.utils.option_utils import update_option_choices
-from griptape_nodes.exe_types.core_types import Parameter
+from griptape_nodes.exe_types.core_types import Parameter, ParameterMessage
 from griptape_nodes.exe_types.node_types import BaseNode
 from griptape_nodes.traits.options import Options
 
@@ -40,6 +40,18 @@ class HuggingFaceModelParameter(ABC):
     def add_input_parameters(self) -> None:
         self._repo_revisions = self.fetch_repo_revisions()
         choices = self.get_choices()
+
+        if not choices:
+            self._node.add_node_element(
+                ParameterMessage(
+                    name="huggingface_repo_parameter_message",
+                    title="Huggingface Model Download Required",
+                    variant="warning",
+                    value=self.get_help_message(),
+                )
+            )
+            return
+
         self._node.add_parameter(
             Parameter(
                 name=self._parameter_name,
@@ -73,12 +85,31 @@ class HuggingFaceModelParameter(ABC):
     def get_repo_revision(self) -> tuple[str, str]:
         value = self._node.get_parameter_value(self._parameter_name)
         if value is None:
-            raise RuntimeError(self.get_help_message())
+            msg = "Model download required!"
+            raise RuntimeError(msg)
         base_repo_id, base_revision = self._key_to_repo_revision(value)
         return base_repo_id, base_revision
 
+    def get_help_message(self) -> str:
+        download_commands = "\n".join([f"  - `{cmd}`" for cmd in self.get_download_commands()])
+        return (
+            "📥 How to download the models:\n"
+            "1. Setup huggingface-cli as per our docs: https://docs.griptapenodes.com/en/stable/how_to/installs/hugging_face/ \n"
+            "2. Download at least one model:\n"
+            f"{download_commands}\n"
+            "3. Save, close, then open the again workflow. (❌ Do not just reload the page)\n"
+            "\n"
+            "✅ If successful, you should see a dropdown with the available models.\n"
+            "❌ If not successful because download fails then check huggingface-cli docs.\n"
+            "❌ If not successful for some other reason then reach out to us on Discord or GitHub.\n"
+            "\n"
+            "Note: Currently this is the only supported method for downloading models. \n"
+            "Hopefully this gets more intuitive (and customizable) soon!\n"
+            "- from ✊📼🍜 with ❤️\n"
+        )
+
     @abstractmethod
     def fetch_repo_revisions(self) -> list[tuple[str, str]]: ...
 
     @abstractmethod
-    def get_help_message(self) -> str: ...
+    def get_download_commands(self) -> list[str]: ...

griptape_nodes-0.36.0/libraries/griptape_nodes_advanced_media_library/diffusers_nodes_library/common/parameters/huggingface_repo_file_parameter.py

@@ -0,0 +1,24 @@
+import logging
+
+from diffusers_nodes_library.common.parameters.huggingface_model_parameter import HuggingFaceModelParameter
+from diffusers_nodes_library.common.utils.huggingface_utils import list_repo_revisions_with_file_in_cache
+from griptape_nodes.exe_types.node_types import BaseNode  # type: ignore[reportMissingImports]
+
+logger = logging.getLogger("diffusers_nodes_library")
+
+
+class HuggingFaceRepoFileParameter(HuggingFaceModelParameter):
+    def __init__(self, node: BaseNode, repo_files: list[tuple[str, str]], parameter_name: str = "model"):
+        super().__init__(node, parameter_name)
+        self._repo_files = repo_files
+        self.refresh_parameters()
+
+    def fetch_repo_revisions(self) -> list[tuple[str, str]]:
+        return [
+            repo_revision
+            for (repo, file) in self._repo_files
+            for repo_revision in list_repo_revisions_with_file_in_cache(repo, file)
+        ]
+
+    def get_download_commands(self) -> list[str]:
+        return [f'huggingface-cli download "{repo}" "{file}"' for (repo, file) in self._repo_files]

griptape_nodes-0.36.0/libraries/griptape_nodes_advanced_media_library/diffusers_nodes_library/common/parameters/huggingface_repo_parameter.py

@@ -0,0 +1,20 @@
+import logging
+
+from diffusers_nodes_library.common.parameters.huggingface_model_parameter import HuggingFaceModelParameter
+from diffusers_nodes_library.common.utils.huggingface_utils import list_repo_revisions_in_cache
+from griptape_nodes.exe_types.node_types import BaseNode  # type: ignore[reportMissingImports]
+
+logger = logging.getLogger("diffusers_nodes_library")
+
+
+class HuggingFaceRepoParameter(HuggingFaceModelParameter):
+    def __init__(self, node: BaseNode, repo_ids: list[str], parameter_name: str = "model"):
+        super().__init__(node, parameter_name)
+        self._repo_ids = repo_ids
+        self.refresh_parameters()
+
+    def fetch_repo_revisions(self) -> list[tuple[str, str]]:
+        return [repo_revision for repo in self._repo_ids for repo_revision in list_repo_revisions_in_cache(repo)]
+
+    def get_download_commands(self) -> list[str]:
+        return [f'huggingface-cli download "{repo}"' for repo in self._repo_ids]

{griptape_nodes-0.35.0 → griptape_nodes-0.36.0}/libraries/griptape_nodes_advanced_media_library/diffusers_nodes_library/common/parameters/log_parameter.py

@@ -49,3 +49,6 @@ class LogParameter:
 
     def append_to_logs(self, text: str) -> None:
         self._node.append_value_to_parameter("logs", text)
+
+    def clear_logs(self) -> None:
+        self._node.publish_update_to_parameter("logs", "")

{griptape_nodes-0.35.0 → griptape_nodes-0.36.0}/libraries/griptape_nodes_advanced_media_library/diffusers_nodes_library/common/utils/torch_utils.py

@@ -22,7 +22,7 @@ def to_human_readable_size(size_in_bytes: float) -> str:
 
 def human_readable_memory_footprint(model: torch.nn.Module) -> str:
     """Return a human-readable memory footprint."""
-    return to_human_readable_size(model.get_memory_footprint())
+    return to_human_readable_size(model.get_memory_footprint())  # type: ignore[reportAttributeAccessIssue]
 
 
 def print_pipeline_memory_footprint(pipe: diffusers.DiffusionPipeline, component_names: list[str]) -> None:

{griptape_nodes-0.35.0 → griptape_nodes-0.36.0}/libraries/griptape_nodes_advanced_media_library/diffusers_nodes_library/pipelines/flux/controlnet/union_flux_control_net_pipeline.py

@@ -84,7 +84,10 @@ class UnionFluxControlNetPipeline(ControlNode):
         with self.log_params.append_profile_to_logs("Loading model"), self.log_params.append_logs_to_logs(logger):
             optimize_flux_pipeline_memory_footprint(pipe)
 
-        with self.log_params.append_profile_to_logs("Configuring flux loras"):
+        with (
+            self.log_params.append_profile_to_logs("Configuring flux loras"),
+            self.log_params.append_logs_to_logs(logger),
+        ):
             self.flux_lora_params.configure_loras(pipe)
 
         num_inference_steps = self.flux_params.get_num_inference_steps()

{griptape_nodes-0.35.0 → griptape_nodes-0.36.0}/libraries/griptape_nodes_advanced_media_library/diffusers_nodes_library/pipelines/flux/controlnet/union_pro_flux_control_net_pipeline.py

@@ -80,7 +80,10 @@ class UnionProFluxControlNetPipeline(ControlNode):
         with self.log_params.append_profile_to_logs("Loading model"), self.log_params.append_logs_to_logs(logger):
             optimize_flux_pipeline_memory_footprint(pipe)
 
-        with self.log_params.append_profile_to_logs("Configuring flux loras"):
+        with (
+            self.log_params.append_profile_to_logs("Configuring flux loras"),
+            self.log_params.append_logs_to_logs(logger),
+        ):
             self.flux_lora_params.configure_loras(pipe)
 
         num_inference_steps = self.flux_params.get_num_inference_steps()

{griptape_nodes-0.35.0 → griptape_nodes-0.36.0}/libraries/griptape_nodes_advanced_media_library/diffusers_nodes_library/pipelines/flux/controlnet/union_pro_two_flux_control_net_pipeline.py

@@ -80,7 +80,10 @@ class UnionProTwoFluxControlNetPipeline(ControlNode):
         with self.log_params.append_profile_to_logs("Loading model"), self.log_params.append_logs_to_logs(logger):
             optimize_flux_pipeline_memory_footprint(pipe)
 
-        with self.log_params.append_profile_to_logs("Configuring flux loras"):
+        with (
+            self.log_params.append_profile_to_logs("Configuring flux loras"),
+            self.log_params.append_logs_to_logs(logger),
+        ):
             self.flux_lora_params.configure_loras(pipe)
 
         num_inference_steps = self.flux_params.get_num_inference_steps()

{griptape_nodes-0.35.0 → griptape_nodes-0.36.0}/libraries/griptape_nodes_advanced_media_library/diffusers_nodes_library/pipelines/flux/flux_fill_pipeline.py

@@ -74,7 +74,10 @@ class FluxFillPipeline(ControlNode):
         with self.log_params.append_profile_to_logs("Loading model"), self.log_params.append_logs_to_logs(logger):
             optimize_flux_pipeline_memory_footprint(pipe)
 
-        with self.log_params.append_profile_to_logs("Configuring flux loras"):
+        with (
+            self.log_params.append_profile_to_logs("Configuring flux loras"),
+            self.log_params.append_logs_to_logs(logger),
+        ):
             self.loras_params.configure_loras(pipe)
 
         num_inference_steps = self.pipe_params.get_num_inference_steps()

{griptape_nodes-0.35.0 → griptape_nodes-0.36.0}/libraries/griptape_nodes_advanced_media_library/diffusers_nodes_library/pipelines/flux/flux_loras_parameter.py

@@ -62,7 +62,7 @@ class FluxLorasParameter:
             lora_path = item["path"]
             msg = f"Loading lora weights: {lora_path}"
             logger.info(msg)
-            state_dict = safetensors.torch.load_file(lora_path)
+            state_dict = safetensors.torch.load_file(lora_path)  # type: ignore[reportAttributeAccessIssue]
             pipe.load_lora_weights(state_dict, adapter_name=item["name"])
 
         if loras_to_load and get_best_device(quiet=True) == torch.device("cuda"):

{griptape_nodes-0.35.0 → griptape_nodes-0.36.0}/libraries/griptape_nodes_advanced_media_library/diffusers_nodes_library/pipelines/flux/flux_pipeline.py

@@ -63,7 +63,10 @@ class FluxPipeline(ControlNode):
         with self.log_params.append_profile_to_logs("Loading model"), self.log_params.append_logs_to_logs(logger):
             optimize_flux_pipeline_memory_footprint(pipe)
 
-        with self.log_params.append_profile_to_logs("Configuring flux loras"):
+        with (
+            self.log_params.append_profile_to_logs("Configuring flux loras"),
+            self.log_params.append_logs_to_logs(logger),
+        ):
             self.loras_params.configure_loras(pipe)
 
         num_inference_steps = self.pipe_params.get_num_inference_steps()

griptape_nodes-0.36.0/libraries/griptape_nodes_advanced_media_library/diffusers_nodes_library/pipelines/flux/peft/__init__.py

@@ -0,0 +1 @@
+"""diffusers_nodes_library/pipelines/flux/peft."""