Erasing-Concepts-In-Diffusion

Runtime error

RohitGandikota commited on Mar 22, 2023

Commit

daebf8f

•

1 Parent(s): 397aa19

added new artists and message after training

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,8 +11,8 @@ model_map = {'Car' : 'models/car.pt',
              'Kilian Eng' : 'models/kilianeng.pt',
              'Thomas Kinkade' : 'models/thomaskinkade.pt',
              'Tyler Edlin' : 'models/tyleredlin.pt',
-             'Kelly McKernan': '/models/kellymckernan.pt',
-             'Rembrandt': '/models/rembrandt.pt' }
 class Demo:

              'Kilian Eng' : 'models/kilianeng.pt',
              'Thomas Kinkade' : 'models/thomaskinkade.pt',
              'Tyler Edlin' : 'models/tyleredlin.pt',
+             'Kelly McKernan': 'models/kellymckernan.pt',
+             'Rembrandt': 'models/rembrandt.pt' }
 class Demo: