Kaynağa Gözat

Make the "no GPU has been detected" message more descriptive

oobabooga 2 yıl önce
ebeveyn
işleme
66256ac1dd
1 değiştirilmiş dosya ile 1 ekleme ve 1 silme
  1. 1 1
      modules/models.py

+ 1 - 1
modules/models.py

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