clamp input (comfyanonymous#5928)

diffus-me · Dec 7, 2024 · fbf68c4 · fbf68c4
1 parent 93477f8
commit fbf68c4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/comfy/ops.py b/comfy/ops.py
@@ -269,7 +269,7 @@ def fp8_linear(self, input):
 
         if scale_input is None:
             scale_input = torch.ones((), device=input.device, dtype=torch.float32)
-            inn = input.reshape(-1, input.shape[2]).to(dtype)
+            inn = torch.clamp(input, min=-448, max=448).reshape(-1, input.shape[2]).to(dtype)
         else:
             scale_input = scale_input.to(input.device)
             inn = (input * (1.0 / scale_input).to(input.dtype)).reshape(-1, input.shape[2]).to(dtype)