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
..
diffusionmodules	Disable autocast for cpu to fix error. Remove unused precision arg. (#518)	2022-09-12 16:55:21 -04:00
distributions	prettified all the code using "blue" at the urging of @tildebyte	2022-08-26 03:15:42 -04:00
encoders	add support for Apple hardware using MPS acceleration	2022-08-31 00:33:23 -04:00
image_degradation	prettified all the code using "blue" at the urging of @tildebyte	2022-08-26 03:15:42 -04:00
losses	prettified all the code using "blue" at the urging of @tildebyte	2022-08-26 03:15:42 -04:00
attention.py	fix opencv and realesrgan dependencies in mac install	2022-09-12 07:07:05 -04:00
ema.py	prettified all the code using "blue" at the urging of @tildebyte	2022-08-26 03:15:42 -04:00
embedding_manager.py	Adding support for .bin files from huggingface concepts (#498)	2022-09-11 15:44:26 -04:00
x_transformer.py	prettified all the code using "blue" at the urging of @tildebyte	2022-08-26 03:15:42 -04:00