Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #85 from JigenD/VRAMutilizationFix

Browse Source 
fix VRAM utilization
This commit is contained in:
Lincoln Stein 2022-08-25 13:47:49 -04:00 committed by GitHub
commit 0e3059728c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
ldm/simplet2i.py
View File

@ -494,6 +494,7 @@ The vast majority of these arguments default to reasonable values.
sd = pl_sd["state_dict"] sd = pl_sd["state_dict"]
model = instantiate_from_config(config.model) model = instantiate_from_config(config.model)
m, u = model.load_state_dict(sd, strict=False) m, u = model.load_state_dict(sd, strict=False)
model.to(self.device)
model.eval() model.eval()
if self.full_precision: if self.full_precision:
print('Using slower but more accurate full-precision math (--full_precision)') print('Using slower but more accurate full-precision math (--full_precision)')