update space

app.py

@@ -257,7 +257,8 @@ def main():
         elif 'cyber' in instruction:
             e_type = 'cyber'
 
-        model = models[e_type].to(device)
+        model = models[e_type]
+        model = model.to(device)
         # model = load_model('text300M', device=device)
         # with torch.no_grad():
         #     new_proj = nn.Linear(1024 * 2, 1024, device=device, dtype=model.wrapped.input_proj.weight.dtype)
@@ -280,7 +281,7 @@ def main():
             latent = latent.to(device)
             text_embeddings_clip = model.cached_model_kwargs(1, dict(texts=[instruction]))
             print("shape of latent: ", latent.clone().unsqueeze(0).shape, "instruction: ", instruction)
-            ref_latent = latent.clone().unsqueeze(0)
+            ref_latent = latent.clone().unsqueeze(0).to(device)
             t_1 = torch.randint(noise_start_t_e_type, noise_start_t_e_type + 1, (1,), device=device).long()
 
             noise_input = diffusion.q_sample(ref_latent, t_1, noise=noise_initial)