Spaces:

Hatman
/

InstantStyle

Running on Zero

Hatman commited on Nov 23, 2024

Commit

f0da3a3

verified ·

1 Parent(s): 88fc2e5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -68,8 +68,8 @@ def create_image(image_pil,
     else:  # If it's already a PIL Image
         style_image = image_pil.convert('RGB')
-    device = next(pipe.parameters()).device
-    generator = torch.Generator(device=device).manual_seed(randomize_seed_fn(seed, True))
     image = pipe(
             width=1024,

     else:  # If it's already a PIL Image
         style_image = image_pil.convert('RGB')
+    generator = torch.Generator().manual_seed(randomize_seed_fn(seed, True))
     image = pipe(
             width=1024,