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

..

modeling

Update on "fixing image_encoder to work with cuda_graphs"

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

utils

Fix incorrect shape in ResizeLongestSide.apply_image_torch.

2023-04-15 10:40:48 -07:00

__init__.py

Initial commit

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

automatic_mask_generator.py

Lint.

2023-04-10 12:02:02 -07:00

build_sam.py

remove direct call to build_sam()

2023-04-07 15:18:56 -07:00

predictor.py

Fix mypy thinking output of SamPredictor.predict is a tensor.

2023-04-12 09:50:56 -07:00