concept ablation

app.py

@@ -6,14 +6,17 @@ from trainer import train_submit, inference
     
 import os
 model_map = {'Van Gogh' : 'models/vangogh_ablation_delta.bin', 
-             'Greg Rutkowski' : 'models/greg_rutkowski_ablation_delta.bin', 
+             'Greg Rutkowski' : 'models/greg_rutkowski_ablation_delta.bin',
+             'R2D2' : 'models/r2d2_delta.bin', 
+             'Grumpy Cat' : 'models/grumpy_cat_delta.bin', 
              }
 
 ORIGINAL_SPACE_ID = 'nupurkmr9/concept-ablation'
 SPACE_ID = os.getenv('SPACE_ID')
 
-SHARED_UI_WARNING = f'''## Attention - the demo requires at least 24GB VRAM for style and object removal, 24 GB VRAM for memorized image removal. Please clone this repository to run on your own machine.   
-<center><a class="duplicate-button" style="display:inline-block" target="_blank" href="https://huggingface.co/spaces/{SPACE_ID}?duplicate=true"><img style="margin-top:0;margin-bottom:0" src="https://img.shields.io/badge/-Duplicate%20Space-blue?labelColor=white&style=flat&logo=data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAABAAAAAQCAYAAAAf8/9hAAAAAXNSR0IArs4c6QAAAP5JREFUOE+lk7FqAkEURY+ltunEgFXS2sZGIbXfEPdLlnxJyDdYB62sbbUKpLbVNhyYFzbrrA74YJlh9r079973psed0cvUD4A+4HoCjsA85X0Dfn/RBLBgBDxnQPfAEJgBY+A9gALA4tcbamSzS4xq4FOQAJgCDwV2CPKV8tZAJcAjMMkUe1vX+U+SMhfAJEHasQIWmXNN3abzDwHUrgcRGmYcgKe0bxrblHEB4E/pndMazNpSZGcsZdBlYJcEL9Afo75molJyM2FxmPgmgPqlWNLGfwZGG6UiyEvLzHYDmoPkDDiNm9JR9uboiONcBXrpY1qmgs21x1QwyZcpvxt9NS09PlsPAAAAAElFTkSuQmCC&logoWidth=14" alt="Duplicate Space"></a></center>. This demo is partly adapted from https://huggingface.co/spaces/baulab/Erasing-Concepts-In-Diffusion.
+SHARED_UI_WARNING = f'''## Attention - the demo requires at least 24GB VRAM for training. Please clone this repository to run on your own machine.   
+<center><a class="duplicate-button" style="display:inline-block" target="_blank" href="https://huggingface.co/spaces/{SPACE_ID}?duplicate=true"><img style="margin-top:0;margin-bottom:0" src="https://img.shields.io/badge/-Duplicate%20Space-blue?labelColor=white&style=flat&logo=data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAABAAAAAQCAYAAAAf8/9hAAAAAXNSR0IArs4c6QAAAP5JREFUOE+lk7FqAkEURY+ltunEgFXS2sZGIbXfEPdLlnxJyDdYB62sbbUKpLbVNhyYFzbrrA74YJlh9r079973psed0cvUD4A+4HoCjsA85X0Dfn/RBLBgBDxnQPfAEJgBY+A9gALA4tcbamSzS4xq4FOQAJgCDwV2CPKV8tZAJcAjMMkUe1vX+U+SMhfAJEHasQIWmXNN3abzDwHUrgcRGmYcgKe0bxrblHEB4E/pndMazNpSZGcsZdBlYJcEL9Afo75molJyM2FxmPgmgPqlWNLGfwZGG6UiyEvLzHYDmoPkDDiNm9JR9uboiONcBXrpY1qmgs21x1QwyZcpvxt9NS09PlsPAAAAAElFTkSuQmCC&logoWidth=14" alt="Duplicate Space"></a></center> 
+This demo is partly adapted from https://huggingface.co/spaces/baulab/Erasing-Concepts-In-Diffusion.
 '''
 
 sys.path.append("concept-ablation-diffusers")
@@ -54,7 +57,7 @@ class Demo:
                     with gr.Column(scale=1):
 
                         self.prompt_input_infr = gr.Text(
-                            placeholder="Enter prompt...",
+                            placeholder="a house in the style of van gogh",
                             label="Prompt",
                             info="Prompt to generate"
                         )

models/grumpy_cat_delta.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2a777d601c1930a88173f4e4a555244f5916576dbf2523277e047c4ede53510c
+size 76685761

models/r2d2_delta.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:22ef614bbb66fa68689b1708f19883f3ad1cda05a90c77a354b9a898593602bc
+size 76685761