Update app.py

app.py

@@ -118,11 +118,6 @@ model = StreamMultiDiffusion(
     seed=opt.seed,
 ).cuda()
 
-model.device = torch.device('cuda')
-model.reset_seed(model.generator, opt.seed)
-model.reset_latent()
-model.prepare()
-
 print(f'[INFO]     Parameters prepared!')
 
 
@@ -390,6 +385,11 @@ def register(state, drawpad):
 
 @spaces.GPU
 def run(state, drawpad):
+    model.device = torch.device('cuda')
+    model.reset_seed(model.generator, opt.seed)
+    model.reset_latent()
+    model.prepare()
+
     state = register(state, drawpad)
     state.is_running = True