5 Commits

Author	SHA1	Message	Date
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
Any-Winter-4079	25d9ccc509	Update model.py	2022-09-11 22:37:45 -04:00
Lincoln Stein	10db192cc4	changes to dogettx optimizations to run on m1 ... * Author @any-winter-4079 * Author @dogettx Thanks to many individuals who contributed time and hardware to benchmarking and debugging these changes.	2022-09-09 09:51:41 -04:00
Lincoln Stein	4f02b72c9c	prettified all the code using "blue" at the urging of @tildebyte	2022-08-26 03:15:42 -04:00
ablattmann	e66308c7f2	add code	2021-12-21 03:23:41 +01:00