fixed (0) in check_cuda()
Browse files
app.py
CHANGED
@@ -209,7 +209,7 @@ def loki_generation(input_text: str,
|
|
209 |
|
210 |
def check_cuda():
|
211 |
if torch.cuda.is_available():
|
212 |
-
return f"GPU Being Used: {torch.cuda.get_device_name
|
213 |
else:
|
214 |
return "No GPU is being used right now."
|
215 |
|
|
|
209 |
|
210 |
def check_cuda():
|
211 |
if torch.cuda.is_available():
|
212 |
+
return f"GPU Being Used: {torch.cuda.get_device_name(0)}"
|
213 |
else:
|
214 |
return "No GPU is being used right now."
|
215 |
|