soaska/my-sd
soaska/my-sd
1
0
Fork 0
Code Issues Pull Requests Actions Packages Activity

add cmd flag hints

Browse Source
This commit is contained in:
lllyasviel 2024-02-23 20:06:08 -08:00
parent 88f395091b
commit d508d8132f
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
ldm_patched/modules/model_management.py
View File

@ -266,10 +266,18 @@ def get_torch_device_name(device):
return "CUDA {}: {}".format(device, torch.cuda.get_device_name(device))
try:
print("Device:", get_torch_device_name(get_torch_device()))
torch_device_name = get_torch_device_name(get_torch_device())
print("Device:", torch_device_name)
except:
torch_device_name = ''
print("Could not pick default device.")
if 'rtx' in torch_device_name.lower():
if not args.pin_shared_memory:
print('Hint: your device supports --pin-shared-memory for potential speed improvements.')
if not args.cuda_malloc:
print('Hint: your device supports --cuda-malloc for potential speed improvements.')
print("VAE dtype:", VAE_DTYPE)
current_loaded_models = []