Charles Hernandez 43c910f431 fixing image_encoder to work with cuda_graphs ...

Summary: the combination of tensors on multiple devices in get_rel_pos
was preventing cuda graphs from correctly optimizing things

Test Plan:

Reviewers:

Subscribers:

Tasks:

Tags:

ghstack-source-id: 2256f130bb8249403710e1048ef69385ff71aed2
Pull Request resolved: https://github.com/facebookresearch/segment-anything/pull/393

2023-05-30 20:37:47 +00:00

..

__init__.py

Initial commit

2023-04-04 22:25:49 -07:00

common.py

Initial commit

2023-04-04 22:25:49 -07:00

image_encoder.py

fixing image_encoder to work with cuda_graphs

2023-05-30 20:37:47 +00:00

mask_decoder.py

Fixed Typo

2023-04-12 11:50:28 -07:00

prompt_encoder.py

Initial commit

2023-04-04 22:25:49 -07:00

sam.py

Fixed some typos in comments

2023-04-07 00:50:18 +08:00

transformer.py

Fixed some typos in comments

2023-04-07 00:50:18 +08:00