"...text-generation-inference.git" did not exist on "6f0f1d70f6871b76d4dd7d6a80a983d3307a0725"
Commit 1bc23a18 authored by Ruilong Li's avatar Ruilong Li
Browse files

minor fix if CUB not supported

parent 1714ba0b
......@@ -208,6 +208,7 @@ def ray_marching(
masks = render_visibility(
alphas,
ray_indices=ray_indices,
packed_info=packed_info,
early_stop_eps=early_stop_eps,
alpha_thre=alpha_thre,
n_rays=rays_o.shape[0],
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment