Spaces:

mfranzon
/

MagicBoard

Runtime error

mfranzon commited on Sep 17, 2022

Commit

e83334f

•

1 Parent(s): ae482b3

fix utils

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -16,7 +16,7 @@ def pipe_image(prompt,
                                                 torch_dtype=torch.float32,
                                                 use_auth_token=TOKEN
                                                 )
-    pipe.to(device)
     pipe.enable_attention_slicing()
     images = pipe(prompt=prompt,
                     init_image=init_image,

                                                 torch_dtype=torch.float32,
                                                 use_auth_token=TOKEN
                                                 )
+    pipe = pipe.to(device)
     pipe.enable_attention_slicing()
     images = pipe(prompt=prompt,
                     init_image=init_image,