mirror of
https://github.com/DrewThomasson/ebook2audiobook.git
synced 2026-01-10 06:18:02 -05:00
...
This commit is contained in:
@@ -548,7 +548,6 @@ class DeviceInstaller():
|
||||
forced_tag = os.environ.get("DEVICE_TAG")
|
||||
if forced_tag:
|
||||
tag_letters = re.match(r"[a-zA-Z]+", forced_tag)
|
||||
print(f'tag_letters: {tag_letters}')
|
||||
if tag_letters:
|
||||
tag_letters = tag_letters.group(0).lower()
|
||||
name = 'cuda' if tag_letters == 'cu' else 'rocm' if tag_letters == 'rocm' else 'jetson' if tag_letters == 'jetson' else 'xpu' if tag_letters == 'xpu' else 'mps' if tag_letters == 'mps' else 'cpu'
|
||||
|
||||
Reference in New Issue
Block a user