update space

app.py

@@ -59,6 +59,7 @@ class Blocks(gr.Blocks):
             config[k] = v
 
         return config
+
 @torch.no_grad()
 def optimize_all(xm, models, initial_noise, noise_start_t, diffusion, latent_model, device, prompt, instruction, rand_seed):
     state = {}
@@ -273,8 +274,11 @@ def main():
     }
     """
     device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+    print("device:", device)
     latent_model = load_model('text300M', device=device)
+    print("loaded latent model")
     xm = load_model('transmitter', device=device)
+    print("loaded transmitter")
     diffusion = diffusion_from_config(load_config('diffusion'))
     freeze_params(xm.parameters())
     models = dict()