Update app.py

app.py

@@ -121,13 +121,14 @@ def process(
     if torch.cuda.is_available():
         model.to("cuda")
         control = control.to("cuda")
-        x_T = x_T.to("cuda")
 
     sampler = SpacedSampler(model, var_type="fixed_small")
     preds = []
     for _ in tqdm(range(num_samples)):
         shape = (1, 4, height // 8, width // 8)
         x_T = torch.randn(shape, device=device, dtype=torch.float32)
+        if torch.cuda.is_available():
+            x_T = x_T.to("cuda")
         if not tile_diffusion:
             samples = sampler.sample_ccsr(
                 steps=steps, t_max=0.6667, t_min=0.3333, shape=shape, cond_img=control,