small note about generating with CPU

2024-08-06 17:40:35 +02:00
parent 9d0c91fb75
commit 5d60957d1e
1 changed files with 1 additions and 1 deletions
@@ -294,7 +294,7 @@ class ImageInference:
                    input_image=None, 
                    mask_image=None,
                    mask_invert=True):
-        # If we make the generation run on CPU here, it will give a different result than GPU
+        # If we make the generation run on CPU here, it will give a different image than GPU
        # IF there are still VRAM issues, try changing this to 'cpu'
        generator = torch.Generator(device='cuda').manual_seed(seed)