Spaces:

sd-concepts-library
/

stable-diffusion-conceptualizer

Runtime error

multimodalart HF staff commited on Sep 15, 2022

Commit

fd45217

•

1 Parent(s): c0391de

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,6 @@ models_list = api.list_models(author="sd-concepts-library", sort="likes", direct
 models = []
 pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", use_auth_token=True, revision="fp16", torch_dtype=torch.float16).to("cuda")
-torch.backends.cudnn.benchmark = True
 def load_learned_embed_in_clip(learned_embeds_path, text_encoder, tokenizer, token=None):
   loaded_learned_embeds = torch.load(learned_embeds_path, map_location="cpu")

 models = []
 pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", use_auth_token=True, revision="fp16", torch_dtype=torch.float16).to("cuda")
 def load_learned_embed_in_clip(learned_embeds_path, text_encoder, tokenizer, token=None):
   loaded_learned_embeds = torch.load(learned_embeds_path, map_location="cpu")