aboutsummaryrefslogtreecommitdiffstats
path: root/modules/devices.py
AgeCommit message (Collapse)AuthorLines
2022-10-25MPS Upscalers Fixbrkirch-0/+4
Get ESRGAN, SCUNet, and SwinIR working correctly on MPS by ensuring memory is contiguous for tensor views before sending to MPS device.
2022-10-25Remove BSRGAN from --use-cpu, add SwinIRbrkirch-1/+1
2022-10-22remove parsing command line from devices.pyAUTOMATIC-9/+5
2022-10-22implement CUDA device selection by IDExtraltodeus-3/+18
2022-10-14Add 'interrogate' and 'all' choices to --use-cpubrkirch-1/+1
* Add 'interrogate' and 'all' choices to --use-cpu * Change type for --use-cpu argument to str.lower, so that choices are case insensitive
2022-10-10--no-half-vaeAUTOMATIC-1/+5
2022-10-04Merge branch 'master' into cpu-cmdline-optbrkirch-0/+10
2022-10-04send all three of GFPGAN's and codeformer's models to CPU memory instead of ↵AUTOMATIC-0/+10
just one for #1283
2022-10-04Add BSRGAN to --add-cpubrkirch-1/+1
2022-10-04Add --use-cpu command line optionbrkirch-3/+2
Remove MPS detection to use CPU for GFPGAN / CodeFormer and add a --use-cpu command line option.
2022-10-04Merge branch 'master' into masterbrkirch-2/+1
2022-10-02initial support for training textual inversionAUTOMATIC-2/+1
2022-09-30When device is MPS, use CPU for GFPGAN insteadbrkirch-1/+1
GFPGAN will not work if the device is MPS, so default to CPU instead.
2022-09-13first attempt to produce crrect seeds in batchAUTOMATIC-0/+10
2022-09-12changes for #294AUTOMATIC-0/+17
2022-09-12 Allow TF32 in CUDA for increased performance #279AUTOMATIC-0/+11
2022-09-11add half() supporrt for CLIP interrogationAUTOMATIC-0/+6
2022-09-11CLIP interrogatorAUTOMATIC-6/+10
2022-09-11Modular device managementAbdullah Barhoum-0/+12