diff --git a/comfy/ldm/modules/diffusionmodules/openaimodel.py b/comfy/ldm/modules/diffusionmodules/openaimodel.py index 4963811a8..4c8d53cac 100644 --- a/comfy/ldm/modules/diffusionmodules/openaimodel.py +++ b/comfy/ldm/modules/diffusionmodules/openaimodel.py @@ -910,4 +910,4 @@ class UNetModel(nn.Module): if self.predict_codebook_ids: return self.id_predictor(h) else: - return self.out(h) \ No newline at end of file + return self.out(h)