19 Commits (e5b4e3f820cd09e751f1d168ab05d606d078a0d9)

Author SHA1 Message Date
brkirch faed465a0b MPS Upscalers Fix
Get ESRGAN, SCUNet, and SwinIR working correctly on MPS by ensuring memory is contiguous for tensor views before sending to MPS device.
3 years ago
brkirch 4c24347e45 Remove BSRGAN from --use-cpu, add SwinIR 3 years ago
AUTOMATIC 50b5504401 remove parsing command line from devices.py 3 years ago
Extraltodeus 57eb54b838
implement CUDA device selection by ID 3 years ago
brkirch fdef8253a4 Add 'interrogate' and 'all' choices to --use-cpu
* Add 'interrogate' and 'all' choices to --use-cpu
* Change type for --use-cpu argument to str.lower, so that choices are case insensitive
3 years ago
AUTOMATIC 7349088d32 --no-half-vae 3 years ago
brkirch e9e2a7ec9a
Merge branch 'master' into cpu-cmdline-opt 3 years ago
AUTOMATIC 6c6ae28bf5 send all three of GFPGAN's and codeformer's models to CPU memory instead of just one for #1283 3 years ago
brkirch 27ddc24fde Add BSRGAN to --add-cpu 3 years ago
brkirch eeab7aedf5 Add --use-cpu command line option
Remove MPS detection to use CPU for GFPGAN / CodeFormer and add a --use-cpu command line option.
3 years ago
brkirch b88e4ea7d6
Merge branch 'master' into master 3 years ago
AUTOMATIC 820f1dc96b initial support for training textual inversion 3 years ago
brkirch bdaa36c844 When device is MPS, use CPU for GFPGAN instead
GFPGAN will not work if the device is MPS, so default to CPU instead.
3 years ago
AUTOMATIC 9d40212485 first attempt to produce crrect seeds in batch 3 years ago
AUTOMATIC c7e0e28ccd changes for #294 3 years ago
AUTOMATIC b70b51cc72 Allow TF32 in CUDA for increased performance #279 3 years ago
AUTOMATIC 8fb9c57ed6 add half() supporrt for CLIP interrogation 3 years ago
AUTOMATIC f194457229 CLIP interrogator 3 years ago
Abdullah Barhoum b5d1af11b7 Modular device management 3 years ago