diff --git a/modelscope/models/cv/nerf_recon_acc/network/nerf.py b/modelscope/models/cv/nerf_recon_acc/network/nerf.py index 64ffae14..8bf711e8 100644 --- a/modelscope/models/cv/nerf_recon_acc/network/nerf.py +++ b/modelscope/models/cv/nerf_recon_acc/network/nerf.py @@ -198,6 +198,8 @@ class VolumeDensity(nn.Module): class VolumeRadiance(nn.Module): def __init__(self, config): + import tinycudann as tcnn + super(VolumeRadiance, self).__init__() self.config = config self.n_dir_dims = 3 diff --git a/modelscope/utils/registry.py b/modelscope/utils/registry.py index e6556d9c..ae81cfc4 100644 --- a/modelscope/utils/registry.py +++ b/modelscope/utils/registry.py @@ -211,4 +211,4 @@ def build_from_cfg(cfg, return obj_cls(**args) except Exception as e: # Normal TypeError does not print class name. - raise type(e)(f'{obj_cls.__name__}: {e}') + raise type(e)(f'{obj_cls.__name__}: {e}') from e