InvokeAI/ldm
Mihai 0bc6779361
Disable autocast for cpu to fix error. Remove unused precision arg. (#518)
When running on just cpu (intel), a call to torch.layer_norm would error with RuntimeError: expected scalar type BFloat16 but found Float
Fix buggy device handling in model.py.
Tested with scripts/dream.py --full_precision on just cpu on intel laptop. Works but slow at ~10s/it.
2022-09-12 16:55:21 -04:00
..
data prettified all the code using "blue" at the urging of @tildebyte 2022-08-26 03:15:42 -04:00
dream Disable autocast for cpu to fix error. Remove unused precision arg. (#518) 2022-09-12 16:55:21 -04:00
gfpgan make gfpgan_model_exists flag available to web interface 2022-09-11 14:47:58 -04:00
models Refactoring simplet2i (#387) 2022-09-05 20:40:10 -04:00
modules Disable autocast for cpu to fix error. Remove unused precision arg. (#518) 2022-09-12 16:55:21 -04:00
generate.py Disable autocast for cpu to fix error. Remove unused precision arg. (#518) 2022-09-12 16:55:21 -04:00
lr_scheduler.py prettified all the code using "blue" at the urging of @tildebyte 2022-08-26 03:15:42 -04:00
simplet2i.py Disabled debug output (#436) 2022-09-11 22:34:35 -04:00
util.py prettified all the code using "blue" at the urging of @tildebyte 2022-08-26 03:15:42 -04:00