diff --git a/ollama/ollama_llm_node.py b/ollama/ollama_llm_node.py index 9cfe2a6..74774cc 100644 --- a/ollama/ollama_llm_node.py +++ b/ollama/ollama_llm_node.py @@ -1,16 +1,7 @@ -import io -import os -import sys import json -import base64 import requests -import torch -import numpy as np -from PIL import Image -from io import BytesIO from typing_extensions import override from comfy_api.latest import ComfyExtension, io -from ..utils.image_utils import tensor_to_base64_string from ..utils.config_utils import get_config_section, get_models_list def _load_config_credentials(config_options=None): diff --git a/ollama/ollama_vlm_node.py b/ollama/ollama_vlm_node.py index 0fd1d1d..b4978e2 100644 --- a/ollama/ollama_vlm_node.py +++ b/ollama/ollama_vlm_node.py @@ -1,13 +1,4 @@ -import io -import os -import sys -import json -import base64 import requests -import torch -import numpy as np -from PIL import Image -from io import BytesIO from typing_extensions import override from comfy_api.latest import ComfyExtension, io from ..utils.image_utils import tensor_to_base64_string