AttributeError: Float8_e5m2

[Huangkangxian]

Traceback (most recent call last):
File "/home/zhujian_test/1-hkx/Code/OpenLRM-main/openlrm/launch.py", line 21, in
from openlrm.runners import REGISTRY_RUNNERS
File "/home/zhujian_test/1-hkx/Code/OpenLRM-main/openlrm/runners/init.py", line 20, in
from .train import *
File "/home/zhujian_test/1-hkx/Code/OpenLRM-main/openlrm/runners/train/init.py", line 16, in
from .lrm import LRMTrainer
File "/home/zhujian_test/1-hkx/Code/OpenLRM-main/openlrm/runners/train/lrm.py", line 21, in
from torchvision.utils import make_grid
File "/home/zhujian_test/.conda/envs/lrm/lib/python3.10/site-packages/torchvision/init.py", line 10, in
from torchvision import _meta_registrations, datasets, io, models, ops, transforms, utils # usort:skip
File "/home/zhujian_test/.conda/envs/lrm/lib/python3.10/site-packages/torchvision/_meta_registrations.py", line 4, in
import torch._custom_ops
File "/home/zhujian_test/.conda/envs/lrm/lib/python3.10/site-packages/torch/_custom_ops.py", line 4, in
from torch._custom_op.impl import (
File "/home/zhujian_test/.conda/envs/lrm/lib/python3.10/site-packages/torch/_custom_op/impl.py", line 9, in
from torchgen.model import FunctionSchema, OperatorName, SchemaKind, BaseType, ListType, BaseTy
File "/home/zhujian_test/.conda/envs/lrm/lib/python3.10/site-packages/torchgen/model.py", line 2354, in
import torchgen.api.ufunc as ufunc
File "/home/zhujian_test/.conda/envs/lrm/lib/python3.10/site-packages/torchgen/api/ufunc.py", line 11, in
import torchgen.api.types as api_types
File "/home/zhujian_test/.conda/envs/lrm/lib/python3.10/site-packages/torchgen/api/types/init.py", line 1, in
from .types import *
File "/home/zhujian_test/.conda/envs/lrm/lib/python3.10/site-packages/torchgen/api/types/types.py", line 100, in
ScalarType.Float8_e5m2: float8_e5m2T,
File "/home/zhujian_test/.conda/envs/lrm/lib/python3.10/enum.py", line 437, in getattr
raise AttributeError(name) from None
AttributeError: Float8_e5m2

How should I solve this error?