mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
0bc6779361
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. |
||
---|---|---|
.. | ||
data | ||
dream | ||
gfpgan | ||
models | ||
modules | ||
generate.py | ||
lr_scheduler.py | ||
simplet2i.py | ||
util.py |