thookham
/

DeOldify

image-colorization

computer-vision

Model card Files Files and versions

DeOldify / deoldify /device_id.py

thookham's picture

Initial commit for Hugging Face sync (Clean History)

e9f9fd3 about 1 month ago

190 Bytes

	from enum import IntEnum

	class DeviceId(IntEnum):
	GPU0 = 0,
	GPU1 = 1,
	GPU2 = 2,
	GPU3 = 3,
	GPU4 = 4,
	GPU5 = 5,
	GPU6 = 6,
	GPU7 = 7,
	XPU = 98,
	CPU = 99