github/ebook2audiobook
1
1
Fork 0
mirror of https://github.com/DrewThomasson/ebook2audiobook.git synced 2026-01-10 06:18:02 -05:00
Code Issues Packages Projects Releases Wiki Activity

...

Browse Source
This commit is contained in:
unknown
2025-12-12 18:13:03 -08:00
parent a2a5ad9133
commit 011d033cdc
2 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
lib/classes/device_installer.py
View File

@@ -455,13 +455,13 @@ class DeviceInstaller():
elif device_info['name'] == devices['MPS']['proc']:
torch_tag_py = f'cp{default_py_major}{default_py_minor}-none'
torchaudio_tag_py = f'cp{default_py_major}{default_py_minor}-cp{default_py_major}{default_py_minor}'
torch_pkg = f'{url}/torch/torch-{torch_version_base}-{torch_tag_py}-{os_env}_{arch}.whl'
torchaudio_pkg = f'{url}/torchaudio/torchaudio-{torch_version_base}-{torchaudio_tag_py}-{os_env}_{arch}.whl'
torch_pkg = f'{url}/cpu/torch-{torch_version_base}-{torch_tag_py}-{os_env}_{arch}.whl'
torchaudio_pkg = f'{url}/cpu/torchaudio-{torch_version_base}-{torchaudio_tag_py}-{os_env}_{arch}.whl'
subprocess.check_call([sys.executable, '-m', 'pip', 'install', '--upgrade', '--no-cache-dir', '--use-pep517', torch_pkg, torchaudio_pkg])
else:
tag_py = f'cp{default_py_major}{default_py_minor}-cp{default_py_major}{default_py_minor}'
torch_pkg = f'{url}/torch/torch-{torch_version_base}+{tag}-{tag_py}-{os_env}_{arch}.whl'
torchaudio_pkg = f'{url}/torchaudio/torchaudio-{torch_version_base}+{tag}-{tag_py}-{os_env}_{arch}.whl'
torch_pkg = f'{url}/{tag}/torch-{torch_version_base}+{tag}-{tag_py}-{os_env}_{arch}.whl'
torchaudio_pkg = f'{url}/{tag}/torchaudio-{torch_version_base}+{tag}-{tag_py}-{os_env}_{arch}.whl'
subprocess.check_call([sys.executable, '-m', 'pip', 'install', '--upgrade', '--no-cache-dir', '--use-pep517', torch_pkg, torchaudio_pkg])
if device_info['name'] == 'cuda':
subprocess.check_call([sys.executable, '-m', 'pip', 'install', '--upgrade', '--no-cache-dir', '--use-pep517', 'deepspeed'])

2
lib/classes/vram_detector.py
View File

@@ -14,7 +14,7 @@ class VRAMDetector:
info = {}
try:
import torch
# ───────────────────────────── Jetson (Unified Memory)
# ───────────────────────────── Jetson (Unified Memory)
if device == 'jetson':
if os.path.exists('/etc/nv_tegra_release'):
try: