Update comfy/float.py

comfy/float.py

@@ -55,8 +55,10 @@ 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 = torch.Generator()
         torch.manual_seed(seed)
+        if(torch.cuda.is_available()):
+            torch.cuda.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))