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")
|