diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2023-01-04 15:09:53 +0300 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2023-01-04 15:09:53 +0300 |
| commit | 642142556d8ecdea9beb86d7618b628b1803ab98 (patch) | |
| tree | 0babb05e3c9a51d97f48e0ead2b94942a70bfe10 /modules | |
| parent | 68fbf4558f9fbf36ca58f92f1fc4681f0ebdf735 (diff) | |
use commandline-supplied cuda device name instead of cuda:0 for safetensors PR that doesn't fix anything
Diffstat (limited to 'modules')
| -rw-r--r-- | modules/sd_models.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/modules/sd_models.py b/modules/sd_models.py index ee918f24..76a89e88 100644 --- a/modules/sd_models.py +++ b/modules/sd_models.py @@ -173,7 +173,7 @@ def read_state_dict(checkpoint_file, print_global_state=False, map_location=None if extension.lower() == ".safetensors":
device = map_location or shared.weight_load_location
if device is None:
- device = "cuda:0" if torch.cuda.is_available() else "cpu"
+ device = devices.get_cuda_device_string() if torch.cuda.is_available() else "cpu"
pl_sd = safetensors.torch.load_file(checkpoint_file, device=device)
else:
pl_sd = torch.load(checkpoint_file, map_location=map_location or shared.weight_load_location)
|