[bugfix] add supports_v1 platform interface (#15417)
Signed-off-by: Joe Runde <Joseph.Runde@ibm.com>
This commit is contained in:
@@ -1666,9 +1666,8 @@ class EngineArgs:
|
||||
_raise_or_fallback(feature_name=name, recommend_to_remove=True)
|
||||
return False
|
||||
|
||||
# No support for device type other than CUDA, AMD (experiemntal) or
|
||||
# TPU (experimental) so far.
|
||||
if not (current_platform.is_cuda_alike() or current_platform.is_tpu()):
|
||||
# Platforms must decide if they can support v1 for this model
|
||||
if not current_platform.supports_v1(model_config=model_config):
|
||||
_raise_or_fallback(
|
||||
feature_name=f"device type={current_platform.device_type}",
|
||||
recommend_to_remove=False)
|
||||
|
||||
Reference in New Issue
Block a user