"docs/git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "a58a4f665b4aa86205fb8c1795e79c331d65bb18"
Unverified Commit 7c4b38ba authored by Nicolas Patry's avatar Nicolas Patry Committed by GitHub
Browse files

Removing `.float()` (`autocast` in fp16 will discard this (I think)). (#495)

parent ab7a78e8
...@@ -333,7 +333,7 @@ class ResnetBlock2D(nn.Module): ...@@ -333,7 +333,7 @@ class ResnetBlock2D(nn.Module):
# make sure hidden states is in float32 # make sure hidden states is in float32
# when running in half-precision # when running in half-precision
hidden_states = self.norm1(hidden_states.float()).type(hidden_states.dtype) hidden_states = self.norm1(hidden_states).type(hidden_states.dtype)
hidden_states = self.nonlinearity(hidden_states) hidden_states = self.nonlinearity(hidden_states)
if self.upsample is not None: if self.upsample is not None:
...@@ -351,7 +351,7 @@ class ResnetBlock2D(nn.Module): ...@@ -351,7 +351,7 @@ class ResnetBlock2D(nn.Module):
# make sure hidden states is in float32 # make sure hidden states is in float32
# when running in half-precision # when running in half-precision
hidden_states = self.norm2(hidden_states.float()).type(hidden_states.dtype) hidden_states = self.norm2(hidden_states).type(hidden_states.dtype)
hidden_states = self.nonlinearity(hidden_states) hidden_states = self.nonlinearity(hidden_states)
hidden_states = self.dropout(hidden_states) hidden_states = self.dropout(hidden_states)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment