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

closes #273, crash on M1 machines

Browse Source
This commit is contained in:
Lincoln Stein 2022-09-01 01:09:56 -04:00
parent 0433b3d625
commit dc30adfbb4
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ldm/simplet2i.py
View File

@ -279,7 +279,7 @@ class T2I:
self._set_sampler()
tic = time.time()
torch.cuda.torch.cuda.reset_peak_memory_stats()
torch.cuda.reset_peak_memory_stats() if self.device == 'cuda' else None
results = list()
try:

2
scripts/dream.py
View File

@ -88,7 +88,7 @@ def main():
tic = time.time()
t2i.load_model()
print(
f'model loaded in', '%4.2fs' % (time.time() - tic)
f'>> model loaded in', '%4.2fs' % (time.time() - tic)
)
if not infile: