meg-huggingface
commited on
Commit
·
5c33832
1
Parent(s):
3d16b0d
Handling more exceptions
Browse files
src/backend/inference_endpoint.py
CHANGED
@@ -37,14 +37,13 @@ def create_endpoint(endpoint_name, repository, framework='pytorch',
|
|
37 |
endpoint.update(repository=repository,
|
38 |
framework=framework, task=task,
|
39 |
accelerator=accelerator,
|
40 |
-
vendor=vendor, region=region,
|
41 |
-
type=type,
|
42 |
instance_size=instance_size,
|
43 |
instance_type=instance_type)
|
44 |
except requests.exceptions.HTTPError as e:
|
45 |
-
# Not enough compute, or
|
46 |
logger.debug("Hit error:")
|
47 |
logger.debug(e)
|
|
|
48 |
endpoint = update_endpoint_exception(endpoint)
|
49 |
except Exception as e:
|
50 |
logger.debug("Hit unaccounted-for error")
|
|
|
37 |
endpoint.update(repository=repository,
|
38 |
framework=framework, task=task,
|
39 |
accelerator=accelerator,
|
|
|
|
|
40 |
instance_size=instance_size,
|
41 |
instance_type=instance_type)
|
42 |
except requests.exceptions.HTTPError as e:
|
43 |
+
# Not enough compute, wrong compute, or quota exceeded
|
44 |
logger.debug("Hit error:")
|
45 |
logger.debug(e)
|
46 |
+
logger.debug("Attempting a different compute.")
|
47 |
endpoint = update_endpoint_exception(endpoint)
|
48 |
except Exception as e:
|
49 |
logger.debug("Hit unaccounted-for error")
|