Cleanup chroma PR.

2025-04-30 20:57:30 -04:00
parent 4ca3d84277
commit 08ff5fa08a
9 changed files with 25 additions and 181 deletions
--- a/comfy/ldm/chroma/layers.py
+++ b/comfy/ldm/chroma/layers.py
@@ -1,7 +1,7 @@
 import torch
 from torch import Tensor, nn

-from .math import attention
+from comfy.ldm.flux.math import attention
 from comfy.ldm.flux.layers import (
    MLPEmbedder,
    RMSNorm,