Commit 2b90b9f
authored
In this PR, we remove `.pin_memory()` in `obj_pos` of `SAM2Base` to resolve and error in MPS. Investigations show that `.pin_memory()` causes an error of `Attempted to set the storage of a tensor on device "cpu" to a storage on different device "mps:0"`, as originally reported in #487.
(close #487)
1 parent 722d1d1 commit 2b90b9f
1 file changed
+2
-4
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
628 | 628 | | |
629 | 629 | | |
630 | 630 | | |
631 | | - | |
632 | | - | |
633 | | - | |
634 | | - | |
| 631 | + | |
| 632 | + | |
635 | 633 | | |
636 | 634 | | |
637 | 635 | | |
| |||
0 commit comments