We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent b54a0d9 commit c068badCopy full SHA for c068bad
src/sagemaker/djl_inference/model.py
@@ -85,7 +85,7 @@ def __new__(
85
config_file = uncompressed_model_data + "/config.json"
86
87
model_type = json.loads(s3.S3Downloader.read_file(config_file)).get("model_type")
88
- cls_to_create = _determine_engine_for_model_type(model_type)
+ cls_to_create = cls if cls is not DJLLargeModel else _determine_engine_for_model_type(model_type)
89
return super(DJLLargeModel, cls).__new__(cls_to_create)
90
91
def __init__(
0 commit comments