init

gradio_dmd.py

@@ -9,7 +9,7 @@ from scheduling_dmd import DMDScheduler
 
 parser = argparse.ArgumentParser()
 parser.add_argument("--unet-path", type='Lykon/dreamshaper-8')
-parser.add_argument("--model-path", type='aaronb/dreamshaper-8-dmd-1kstep')
+parser.add_argument("--model-path", type='aaronb/dreamshaper-8-dmd-kl-only-6kstep')
 args = parser.parse_args()
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")