pookiefoof Claude Opus 4.7 (1M context) commited on
Commit
d24433c
·
1 Parent(s): d164652

Use float16 instead of bfloat16 to match reference inference

Browse files

torch_cluster.fps (called inside the VAE encoder for FPS sub-sampling)
has CUDA kernels for fp16/fp32/fp64 but not bf16, so the bf16 surface
tensor raised NotImplementedError: "_" not implemented for 'BFloat16'.
The reference scripts/inference_detailgen3d.py uses dtype=torch.float16,
which works.

Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -23,7 +23,7 @@ sys.path.append(os.path.dirname(os.path.abspath(__file__)))
23
  # Constants
24
  MAX_SEED = np.iinfo(np.int32).max
25
  TMP_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), "tmp")
26
- DTYPE = torch.bfloat16
27
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
28
 
29
 
 
23
  # Constants
24
  MAX_SEED = np.iinfo(np.int32).max
25
  TMP_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), "tmp")
26
+ DTYPE = torch.float16
27
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
28
 
29