devices fix

app.py

@@ -75,11 +75,11 @@ def run_all(prompt, steps, n_images, weight, clip_guided):
         
         for prompt in prompts:
             txt, weight = parse_prompt(prompt)
-            target_embeds.append(clip_model.encode_text(clip.tokenize(txt).to(device)).float())
+            target_embeds.append(clip_model.encode_text(clip.tokenize(txt).to('cuda')).float())
             weights.append(weight)
         
         target_embeds = torch.cat(target_embeds)
-        weights = torch.tensor(weights, device=device)
+        weights = torch.tensor(weights, device='cuda')
         if weights.sum().abs() < 1e-3:
             raise RuntimeError('The weights must not sum to 0.')
         weights /= weights.sum().abs()