From c612d348ba301002d7659e860b5b72890d906c7e Mon Sep 17 00:00:00 2001 From: Lucain Pouget Date: Fri, 3 Nov 2023 15:47:23 +0100 Subject: [PATCH] Update SpaceHardware enum --- src/huggingface_hub/_space_api.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/huggingface_hub/_space_api.py b/src/huggingface_hub/_space_api.py index 711ad50070..ddfcd041d6 100644 --- a/src/huggingface_hub/_space_api.py +++ b/src/huggingface_hub/_space_api.py @@ -61,8 +61,11 @@ class SpaceHardware(str, Enum): CPU_UPGRADE = "cpu-upgrade" T4_SMALL = "t4-small" T4_MEDIUM = "t4-medium" + ZERO_A10G = "zero-a10g" A10G_SMALL = "a10g-small" A10G_LARGE = "a10g-large" + A10G_LARGEX2 = "a10g-largex2" + A10G_LARGEX4 = "a10g-largex4" A100_LARGE = "a100-large"