in shap_e/models/nerstf/renderer.py [0:0]
def __init__(
self,
sdf: Optional[Model],
tf: Optional[Model],
nerstf: Optional[Model],
void: NeRFModel,
volume: Volume,
grid_size: int,
n_coarse_samples: int,
n_fine_samples: int,
importance_sampling_options: Optional[Dict[str, Any]] = None,
separate_shared_samples: bool = False,
texture_channels: Sequence[str] = ("R", "G", "B"),
channel_scale: Sequence[float] = (255.0, 255.0, 255.0),
ambient_color: Union[float, Tuple[float]] = BASIC_AMBIENT_COLOR,
diffuse_color: Union[float, Tuple[float]] = BASIC_DIFFUSE_COLOR,
specular_color: Union[float, Tuple[float]] = 0.0,
output_srgb: bool = True,
device: torch.device = torch.device("cuda"),
**kwargs,