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

do not use autocast for diffusers

Browse Source 
- All tensors in diffusers code path are now set explicitly to
  float32 or float16, depending on the --precision flag.
- autocast is still used in the ckpt path, since it is being
  deprecated.
This commit is contained in:
Lincoln Stein
2023-01-16 19:32:06 -05:00
parent 563196bd03
commit 7e8f364d8d
11 changed files with 46 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
ldm/invoke/model_manager.py
View File

@ -761,7 +761,7 @@ class ModelManager(object):
for model in legacy_locations:
source = models_dir /model
if source.exists():
print(f'DEBUG: Moving {models_dir / model} into hub')
print(f'** Moving {models_dir / model} into hub')
move(models_dir / model, hub)
# anything else gets moved into the diffusers directory