aboutsummaryrefslogtreecommitdiffstats
path: root/modules/devices.py
Commit message (Expand)AuthorAgeFilesLines
* Fix potential bugsKohaku-Blueleaf2024-01-291-2/+7
* Avoid exceptions to be silencedKohaku-Blueleaf2024-01-201-6/+5
* Avoid early disableKohaku-Blueleaf2024-01-201-0/+4
* Fix nested manual castKohaku-Blueleaf2024-01-181-1/+5
* rearrange if-statements for cpuKohaku-Blueleaf2024-01-091-3/+3
* Apply the correct behavior of precision='full'Kohaku-Blueleaf2024-01-091-4/+7
* Revert "Apply correct inference precision implementation"Kohaku-Blueleaf2024-01-091-33/+9
* Apply correct inference precision implementationKohaku-Blueleaf2024-01-091-9/+33
* linting and debugsKohaku-Blueleaf2024-01-091-6/+6
* Fix bugs when arg dtype doesn't matchKohakuBlueleaf2024-01-091-15/+10
* improve efficiency and support more deviceKohaku-Blueleaf2024-01-091-17/+43
* change import statements for #14478AUTOMATIC11112023-12-311-2/+2
* Add utility to inspect a model's parameters (to get dtype/device)Aarni Koskela2023-12-311-1/+2
* Merge branch 'dev' into test-fp8Kohaku-Blueleaf2023-12-031-0/+13
|\
| * Merge pull request #14171 from Nuullll/ipexAUTOMATIC11112023-12-021-0/+13
| |\
| | * Disable ipex autocast due to its bad perfNuullll2023-12-021-7/+13
| | * Initial IPEX supportNuullll2023-11-301-2/+9
* | | Merge branch 'dev' into test-fp8Kohaku-Blueleaf2023-12-021-1/+1
|\| |
| * | Merge pull request #14131 from read-0nly/patch-1AUTOMATIC11112023-12-021-1/+1
| |\ \ | | |/ | |/|
| | * Update devices.pyobsol2023-11-281-1/+1
* | | Better namingKohaku-Blueleaf2023-11-191-3/+3
* | | Use options instead of cmd_argsKohaku-Blueleaf2023-11-191-11/+14
* | | Add MPS manual castKohakuBlueleaf2023-10-281-1/+5
* | | ManualCast for 10/16 series gpuKohaku-Blueleaf2023-10-281-6/+51
* | | Add CPU fp8 supportKohaku-Blueleaf2023-10-231-1/+5
|/ /
* | fix for crash when running #12924 without --device-idAUTOMATIC11112023-09-091-1/+1
* | More accurate check for enabling cuDNN benchmark on 16XX cardscatboxanon2023-08-311-1/+2
|/
* split shared.py into multiple files; should resolve all circular reference im...AUTOMATIC11112023-08-091-9/+1
* rework RNG to use generators instead of generating noises beforehandAUTOMATIC11112023-08-091-79/+2
* rework torchsde._brownian.brownian_interval replacement to use device.randn_l...AUTOMATIC11112023-08-031-6/+38
* add NV option for Random number generator source setting, which allows to gen...AUTOMATIC11112023-08-021-2/+37
* Fix MPS cache cleanupAarni Koskela2023-07-111-2/+3
* added torch.mps.empty_cache() to torch_gc()AUTOMATIC11112023-07-081-0/+3
* Remove a bunch of unused/vestigial codeAarni Koskela2023-06-051-7/+0
* run basic torch calculation at startup in parallel to reduce the performance ...AUTOMATIC2023-05-211-0/+18
* ruff auto fixesAUTOMATIC2023-05-101-1/+1
* rename CPU RNG to RNG source in settings, add infotext and parameters copypas...AUTOMATIC2023-04-291-2/+2
* Option to use CPU for random number generation.Deciare2023-04-191-2/+6
* Refactor Mac specific code to a separate filebrkirch2023-02-011-45/+7
* Refactor MPS fixes to CondFuncbrkirch2023-02-011-36/+14
* MPS fix is still needed :(brkirch2023-02-011-0/+3
* Merge pull request #7309 from brkirch/fix-embeddingsAUTOMATIC11112023-01-281-3/+8
|\
| * Remove MPS fix no longer needed for PyTorchbrkirch2023-01-281-3/+0
| * Refactor conditional casting, fix upscalersbrkirch2023-01-281-0/+8
* | clarify the option to disable NaN check.AUTOMATIC2023-01-271-0/+2
* | remove the need to place configs near modelsAUTOMATIC2023-01-271-4/+8
|/
* Add UI setting for upcasting attention to float32brkirch2023-01-251-1/+5
* Add option for float32 sampling with float16 UNetbrkirch2023-01-251-0/+2
* Merge pull request #6922 from brkirch/cumsum-fixAUTOMATIC11112023-01-191-4/+7
|\
| * Fix cumsum for MPS in newer torchbrkirch2023-01-181-4/+7