mirror of
https://github.com/jantic/DeOldify.git
synced 2026-09-01 15:36:32 +08:00
Jason Antic
13 lines
176 B
Python
13 lines
176 B
Python
from enum import IntEnum
|
|
|
|
class DeviceId(IntEnum):
|
|
GPU0 = 0,
|
|
GPU1 = 1,
|
|
GPU2 = 2,
|
|
GPU3 = 3,
|
|
GPU4 = 4,
|
|
GPU5 = 5,
|
|
GPU6 = 6,
|
|
GPU7 = 7,
|
|
CPU = 99
|