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

renam LastLayer for FinalLayer

Browse Source
This commit is contained in:
David Bertoin
2025-10-13 11:40:52 +00:00
committed by DavidBert
parent 3c60c9230e
commit de1ceaf07a
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/diffusers/models/transformers/transformer_photon.py
View File

@@ -428,7 +428,7 @@ class PhotonBlock(nn.Module):
return img
class LastLayer(nn.Module):
class FinalLayer(nn.Module):
r"""
Final projection layer with adaptive LayerNorm modulation.
@@ -634,7 +634,7 @@ class PhotonTransformer2DModel(ModelMixin, ConfigMixin, AttentionMixin):
]
)
self.final_layer = LastLayer(self.hidden_size, 1, self.out_channels)
self.final_layer = FinalLayer(self.hidden_size, 1, self.out_channels)
self.gradient_checkpointing = False