Fix ruff.

This commit is contained in:
comfyanonymous 2026-01-04 18:34:58 -05:00
parent 4dc8dad3bd
commit 022c307f38
3 changed files with 5 additions and 4 deletions

View File

@ -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:

View File

@ -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)

View File

@ -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/"