Update comfy/float.py

comfy/float.py

@@ -62,7 +62,7 @@ def stochastic_rounding(value, dtype, seed=0):
         slice_size = max(1, round(value.shape[0] / num_slices))
         with torch.no_grad():
             for i in range(0, value.shape[0], slice_size):
-                output[i:i+slice_size].copy_(manual_stochastic_round_to_float8(value[i:i+slice_size], dtype, generator=generator))
+                output[i:i+slice_size].copy_(manual_stochastic_round_to_float8(value[i:i+slice_size], dtype))
         return output
 
     return value.to(dtype=dtype)