lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-01-26 19:09:45 +00:00
Code Issues Packages Projects Releases Wiki 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
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 = []