Spaces:

tincri
/

AutoStyleGAN

Sleeping

tincri commited on Apr 28, 2025

Commit

0286875

1 Parent(s): 055ba7e

Fix #10 app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -227,7 +227,7 @@ if __name__ == '__main__':
     checkpoint_path = 'iter/12500_nets_ema.ckpt'
     img_size = 128
     style_dim = 64
-    num_domains = 3
     device = 'cuda' if torch.cuda.is_available() else 'cpu'
     try:

     checkpoint_path = 'iter/12500_nets_ema.ckpt'
     img_size = 128
     style_dim = 64
+    num_domains = 2
     device = 'cuda' if torch.cuda.is_available() else 'cpu'
     try: