michael/stable-diffusion-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Typo StableDiffuion

Browse source 
Small typo StableDiffuionModelHijack() => StableDiffusionModelHijack()
This commit is contained in:
orionaskatu 2022-08-28 15:46:44 +02:00 committed by GitHub
parent 93e7dbaa71
commit e5480e3688
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
webui.py
View file

@ -519,7 +519,7 @@ if os.path.exists(cmd_opts.gfpgan_dir):
print(traceback.format_exc(), file=sys.stderr) print(traceback.format_exc(), file=sys.stderr)
class StableDiffuionModelHijack: class StableDiffusionModelHijack:
ids_lookup = {} ids_lookup = {}
word_embeddings = {} word_embeddings = {}
word_embeddings_checksums = {} word_embeddings_checksums = {}
@ -1328,7 +1328,7 @@ sd_model = load_model_from_config(sd_config, cmd_opts.ckpt)
device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu") device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
sd_model = (sd_model if cmd_opts.no_half else sd_model.half()).to(device) sd_model = (sd_model if cmd_opts.no_half else sd_model.half()).to(device)
model_hijack = StableDiffuionModelHijack() model_hijack = StableDiffusionModelHijack()
model_hijack.hijack(sd_model) model_hijack.hijack(sd_model)
demo = gr.TabbedInterface( demo = gr.TabbedInterface(