1
0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00

make quality + style

This commit is contained in:
David Bertoin
2025-10-13 13:09:00 +00:00
committed by DavidBert
parent 8ee17d20b3
commit bb36735379
2 changed files with 2 additions and 3 deletions

View File

@@ -340,7 +340,7 @@ if __name__ == "__main__":
default=DEFAULT_RESOLUTION,
help="Target resolution for the model (256, 512, or 1024). Affects the transformer's sample_size.",
)
parser.add_argument(
"--shift",
type=float,

View File

@@ -227,6 +227,7 @@ class QKNorm(torch.nn.Module):
k = self.key_norm(k)
return q.to(v), k.to(v)
class Modulation(nn.Module):
r"""
Modulation network that generates scale, shift, and gating parameters.
@@ -339,8 +340,6 @@ class PhotonBlock(nn.Module):
self.modulation = Modulation(hidden_size)
def forward(
self,
img: Tensor,