Update comfy/float.py

comfy/float.py

@@ -55,15 +55,14 @@ def stochastic_rounding(value, dtype, seed=0):
     if dtype == torch.bfloat16:
         return value.to(dtype=torch.bfloat16)
     if dtype == torch.float8_e4m3fn or dtype == torch.float8_e5m2:
-        generator = torch.Generator()
-        generator.manual_seed(seed)
-        generator.to("cuda")
+        #generator = torch.Generator()
+        torch.manual_seed(seed)
         output = torch.empty_like(value, dtype=dtype)
         num_slices = max(1, (value.numel() / (4096 * 4096)))
         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)