Skip to content

Commit 66256ac

Browse files
committed
Make the "no GPU has been detected" message more descriptive
1 parent 67d6247 commit 66256ac

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

modules/models.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -98,7 +98,7 @@ def load_model(model_name):
9898
command = "AutoModelForCausalLM.from_pretrained"
9999
params = ["low_cpu_mem_usage=True"]
100100
if not shared.args.cpu and not torch.cuda.is_available():
101-
print("Warning: no GPU has been detected.\nFalling back to CPU mode.\n")
101+
print("Warning: torch.cuda.is_available() returned False.\nThis means that no GPU has been detected.\nFalling back to CPU mode.\n")
102102
shared.args.cpu = True
103103

104104
if shared.args.cpu:

0 commit comments

Comments
 (0)