Skip to content

Commit 027dbd5

Browse files
committed
make quality + style
1 parent 9ad5720 commit 027dbd5

File tree

2 files changed

+2
-3
lines changed

2 files changed

+2
-3
lines changed

scripts/convert_photon_to_diffusers.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -340,7 +340,7 @@ def main(args):
340340
default=DEFAULT_RESOLUTION,
341341
help="Target resolution for the model (256, 512, or 1024). Affects the transformer's sample_size.",
342342
)
343-
343+
344344
parser.add_argument(
345345
"--shift",
346346
type=float,

src/diffusers/models/transformers/transformer_photon.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -227,6 +227,7 @@ def forward(self, q: Tensor, k: Tensor, v: Tensor) -> tuple[Tensor, Tensor]:
227227
k = self.key_norm(k)
228228
return q.to(v), k.to(v)
229229

230+
230231
class Modulation(nn.Module):
231232
r"""
232233
Modulation network that generates scale, shift, and gating parameters.
@@ -339,8 +340,6 @@ def __init__(
339340

340341
self.modulation = Modulation(hidden_size)
341342

342-
343-
344343
def forward(
345344
self,
346345
img: Tensor,

0 commit comments

Comments
 (0)