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:
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
Reference in New Issue
Block a user