Fix linting issues
This commit is contained in:
+2
-3
@@ -1,8 +1,8 @@
|
|||||||
# SPDX-License-Identifier: AGPL-3.0-or-later
|
# SPDX-License-Identifier: AGPL-3.0-or-later
|
||||||
# Copyright (C) 2025 Philipp Emanuel Weidmann <pew@worldwidemann.com>
|
# Copyright (C) 2025 Philipp Emanuel Weidmann <pew@worldwidemann.com>
|
||||||
|
|
||||||
import os
|
|
||||||
import math
|
import math
|
||||||
|
import os
|
||||||
import sys
|
import sys
|
||||||
import time
|
import time
|
||||||
import warnings
|
import warnings
|
||||||
@@ -41,7 +41,6 @@ from .utils import (
|
|||||||
get_trial_parameters,
|
get_trial_parameters,
|
||||||
load_prompts,
|
load_prompts,
|
||||||
print,
|
print,
|
||||||
empty_cache,
|
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
@@ -96,7 +95,7 @@ def run():
|
|||||||
elif is_npu_available():
|
elif is_npu_available():
|
||||||
print(f"CANN version: [bold]{torch.version.cann}[/]")
|
print(f"CANN version: [bold]{torch.version.cann}[/]")
|
||||||
elif torch.backends.mps.is_available():
|
elif torch.backends.mps.is_available():
|
||||||
print(f"GPU type: [bold]Apple Metal (MPS)[/]")
|
print("GPU type: [bold]Apple Metal (MPS)[/]")
|
||||||
else:
|
else:
|
||||||
print(
|
print(
|
||||||
"[bold yellow]No GPU or other accelerator detected. Operations will be slow.[/]"
|
"[bold yellow]No GPU or other accelerator detected. Operations will be slow.[/]"
|
||||||
|
|||||||
Reference in New Issue
Block a user