Merge pull request #85 from JigenD/VRAMutilizationFix

fix VRAM utilization
This commit is contained in:
Lincoln Stein
2022-08-25 13:47:49 -04:00
committed by GitHub

View File

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