monkey_patch _compute_timestep_embedding

app.py

@@ -62,7 +62,7 @@ def get_timestep_embedding(
         emb = torch.nn.functional.pad(emb, (0, 1, 0, 0))
     return emb
  
-def _compute_timestep_embedding(self, timestep: torch.Tensor, dtype: torch.dtype) -> torch.Tensor:
+def _compute_timestep_embedding(timestep: torch.Tensor, dtype: torch.dtype) -> torch.Tensor:
         return self.time_in(
             get_timestep_embedding(
                 timesteps=timestep,