From 022c307f38a7bfb2c92a5a9e885f9ddb81e32b52 Mon Sep 17 00:00:00 2001 From: comfyanonymous Date: Sun, 4 Jan 2026 18:34:58 -0500 Subject: [PATCH] Fix ruff. --- comfy/ldm/lightricks/vae/causal_audio_autoencoder.py | 5 +++-- comfy/text_encoders/lt.py | 2 +- pyproject.toml | 2 +- 3 files changed, 5 insertions(+), 4 deletions(-) diff --git a/comfy/ldm/lightricks/vae/causal_audio_autoencoder.py b/comfy/ldm/lightricks/vae/causal_audio_autoencoder.py index b2043c28c..f12b9bb53 100644 --- a/comfy/ldm/lightricks/vae/causal_audio_autoencoder.py +++ b/comfy/ldm/lightricks/vae/causal_audio_autoencoder.py @@ -6,6 +6,7 @@ from typing import Optional from enum import Enum from .pixel_norm import PixelNorm import comfy.ops +import logging ops = comfy.ops.disable_weight_init @@ -400,9 +401,9 @@ def make_attn(in_channels, attn_type="vanilla", norm_type="group"): attn_type = AttentionType.str_to_enum(attn_type) if attn_type != AttentionType.NONE: - print(f"making attention of type '{attn_type.value}' with {in_channels} in_channels") + logging.info(f"making attention of type '{attn_type.value}' with {in_channels} in_channels") else: - print(f"making identity attention with {in_channels} in_channels") + logging.info(f"making identity attention with {in_channels} in_channels") match attn_type: case AttentionType.VANILLA: diff --git a/comfy/text_encoders/lt.py b/comfy/text_encoders/lt.py index abd8e5d6e..2c2d453e8 100644 --- a/comfy/text_encoders/lt.py +++ b/comfy/text_encoders/lt.py @@ -113,7 +113,7 @@ class LTXAVTEModel(torch.nn.Module): sdo = comfy.utils.state_dict_prefix_replace(sd, {"text_embedding_projection.aggregate_embed.weight": "text_embedding_projection.weight", "model.diffusion_model.video_embeddings_connector.": "video_embeddings_connector.", "model.diffusion_model.audio_embeddings_connector.": "audio_embeddings_connector."}, filter_keys=True) if len(sdo) == 0: sdo = sd - print(sdo.keys()) + return self.load_state_dict(sdo, strict=False) diff --git a/pyproject.toml b/pyproject.toml index 60378de1e..a7d159be9 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -3,7 +3,7 @@ name = "ComfyUI" version = "0.7.0" readme = "README.md" license = { file = "LICENSE" } -requires-python = ">=3.9" +requires-python = ">=3.10" [project.urls] homepage = "https://www.comfy.org/"