Remove comments no longer appropriate (open-mmlab#716)

Remove comments no longer appropriate. There were casting operations before, they are now gone.
aravind-h-v · Oct 4, 2022 · 6b22192 · 6b22192
1 parent 215bb40
commit 6b22192
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 6 deletions.
diff --git a/src/diffusers/models/resnet.py b/src/diffusers/models/resnet.py
@@ -335,8 +335,6 @@ def __init__(
     def forward(self, x, temb):
         hidden_states = x
 
-        # make sure hidden states is in float32
-        # when running in half-precision
         hidden_states = self.norm1(hidden_states)
         hidden_states = self.nonlinearity(hidden_states)
 
@@ -353,8 +351,6 @@ def forward(self, x, temb):
             temb = self.time_emb_proj(self.nonlinearity(temb))[:, :, None, None]
             hidden_states = hidden_states + temb
 
-        # make sure hidden states is in float32
-        # when running in half-precision
         hidden_states = self.norm2(hidden_states)
         hidden_states = self.nonlinearity(hidden_states)
 

diff --git a/src/diffusers/models/unet_2d_condition.py b/src/diffusers/models/unet_2d_condition.py
@@ -313,8 +313,6 @@ def forward(
                     hidden_states=sample, temb=emb, res_hidden_states_tuple=res_samples, upsample_size=upsample_size
                 )
         # 6. post-process
-        # make sure hidden states is in float32
-        # when running in half-precision
         sample = self.conv_norm_out(sample)
         sample = self.conv_act(sample)
         sample = self.conv_out(sample)