Spaces:

tetrisd
/

Diffusion-Attentive-Attribution-Maps

Runtime error

tetrisd commited on Nov 10, 2022

Commit

f4dd424

•

1 Parent(s): 4e930c1

Update to use Stable Diffusion v1.5

Files changed (1) hide show

app.py CHANGED Viewed

@@ -24,7 +24,7 @@ if not os.environ.get('NO_DOWNLOAD_SPACY'):
     spacy.cli.download('en_core_web_sm')
-model_id = "CompVis/stable-diffusion-v1-4"
 device = "cuda"
 gen = torch.Generator(device='cuda')

     spacy.cli.download('en_core_web_sm')
+model_id = "runwayml/stable-diffusion-v1-5"
 device = "cuda"
 gen = torch.Generator(device='cuda')