Update sampling.py

2024-03-04 19:55:44 +02:00
parent 98fdca7613
commit cb8ebb1d80
1 changed files with 1 additions and 1 deletions
@@ -533,7 +533,7 @@ def gaussian_weights(tile_width, tile_height, nbatches):
               for y in range(latent_height)]
    
    if comfy.model_management.is_device_mps(device):
-        weights = np.outer(y_probs, x_probs, dtype=np.float32)
+        weights = np.outer(y_probs, x_probs).astype(np.float32)
    else:
        weights = np.outer(y_probs, x_probs)