Update to SD 2.1

app.py

@@ -54,11 +54,12 @@ def get_args():
     model_id_map = {
         'v1': 'runwayml/stable-diffusion-v1-5',
         'v2-base': 'stabilityai/stable-diffusion-2-base',
-        'v2-large': 'stabilityai/stable-diffusion-2'
+        'v2-large': 'stabilityai/stable-diffusion-2',
+        'v2-1-base': 'stabilityai/stable-diffusion-2-1-base'
     }
 
     parser = argparse.ArgumentParser()
-    parser.add_argument('--model', '-m', type=str, default='v2-base', choices=list(model_id_map.keys()), help="which diffusion model to use")
+    parser.add_argument('--model', '-m', type=str, default='v2-1-base', choices=list(model_id_map.keys()), help="which diffusion model to use")
     parser.add_argument('--seed', '-s', type=int, default=0, help="the random seed")
     parser.add_argument('--port', '-p', type=int, default=8080, help="the port to launch the demo")
     parser.add_argument('--no-cuda', action='store_true', help="Use CPUs instead of GPUs")
@@ -168,6 +169,8 @@ def main():
         md = '''# DAAM: Attention Maps for Interpreting Stable Diffusion
         Check out the **new** paper (2022/12/7): [What the DAAM: Interpreting Stable Diffusion Using Cross Attention](http://arxiv.org/abs/2210.04885).
         See our (much cleaner) [DAAM codebase](https://github.com/castorini/daam) on GitHub.
+
+        Current model: Stable Diffusion 2.1 base.
         '''
         gr.Markdown(md)