StreamMultiDiffusion

Runtime error

ironjr commited on Apr 12

Commit

934abd0

•

1 Parent(s): 96bf51f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -118,7 +118,7 @@ model = StreamMultiDiffusion(
     seed=opt.seed,
 ).cuda()
 model.reset_seed(model.generator, opt.seed)
 model.reset_latent()
 model.prepare()

     seed=opt.seed,
 ).cuda()
+model.device = torch.device('cuda')
 model.reset_seed(model.generator, opt.seed)
 model.reset_latent()
 model.prepare()