Charles Hernandez 51bc7a2e0b Update on "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-poisoned]

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

Update on "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