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