Tang, Sheng-Kai comments

Repositories
Issues
Comments

Results 1 comments of


                                            Tang, Sheng-Kai

run with "mps" is error:Cannot convert a MPS Tensor to float64 dtype as the MPS framework doesn't support float64. Please use float32 instead.

> > Replace this line: > > > > https://github.com/facebookresearch/segment-anything/blob/6fdee8f2727f4506cfbbe553e23b895e27956588/segment_anything/automatic_mask_generator.py#L277 > > > > with > > `in_points = torch.from_numpy(transformed_points.astype(np.float32)).to(self.predictor.device)` > > to make it work in MPS. > >...