Update comfy/float.py

comfy/float.py

@@ -55,7 +55,7 @@ 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(device='cuda' if torch.cuda.is_available() else 'cpu')
+        #generator = torch.Generator(device='cuda' if torch.cuda.is_available() else 'cpu')
         torch.manual_seed(seed)
         if(torch.cuda.is_available()):
             torch.cuda.manual_seed(seed)
@@ -64,7 +64,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)