Spaces:

snap-research
/

weights2weights

Running on Zero

amildravid4292 commited on Jul 22, 2024

Commit

37a1a22

verified ·

1 Parent(s): c2aa185

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -209,7 +209,9 @@ def edit_inference(net, prompt, negative_prompt, guidance_scale, ddim_steps, see
     mean.to(device)
     std.to(device)
     v.to(device)
     young.to(device)
     pointy.to(device)
     wavy.to(device)
     thick.to(device)

     mean.to(device)
     std.to(device)
     v.to(device)
+    print(young.device)
     young.to(device)
+    print(young.device)
     pointy.to(device)
     wavy.to(device)
     thick.to(device)