Skip to content

Commit 5902ab0

Browse files
committed
disable refit for py313 due to torch 2.7 issue
1 parent 7adc997 commit 5902ab0

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

py/torch_tensorrt/_features.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@
3737
_TORCHTRT_RT_AVAIL = _TS_FE_AVAIL or os.path.isfile(linked_file_runtime_full_path)
3838
_DYNAMO_FE_AVAIL = version.parse(sanitized_torch_version()) >= version.parse("2.1.dev")
3939
_FX_FE_AVAIL = True
40-
_REFIT_AVAIL = True
40+
_REFIT_AVAIL = version.parse(sys.version.split()[0]) < version.parse("3.13")
4141

4242
ENABLED_FEATURES = FeatureSet(
4343
_TS_FE_AVAIL, _TORCHTRT_RT_AVAIL, _DYNAMO_FE_AVAIL, _FX_FE_AVAIL, _REFIT_AVAIL

0 commit comments

Comments
 (0)