Update transformer/transformer.py

transformer/transformer.py

@@ -112,7 +112,7 @@ class TimestepEmbeddings(nn.Module):
         return timesteps_emb
 
 
-class EspressoMMDiTModel(SD3Transformer2DModel):
+class NitroMMDiTModel(SD3Transformer2DModel):
     _supports_gradient_checkpointing = True
 
     @register_to_config