scott/test2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

maybe fix

Browse Source
This commit is contained in:
Scott Register
2025-07-26 08:47:50 -07:00
parent e195d23584
commit 7852303b40
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
vr180_matting/sam2_wrapper.py
View File

@@ -67,10 +67,12 @@ class SAM2VideoMatting:
# Enable memory optimizations # Enable memory optimizations
if self.memory_offload: if self.memory_offload:
self.predictor.fill_hole_area = 8 # SAM2 has different memory optimization options
pass # Memory offloading is handled by SAM2 internally
if self.fp16 and self.device == "cuda": if self.fp16 and self.device == "cuda":
self.predictor.model.half() # SAM2 handles FP16 internally, no need to manually convert
pass
except Exception as e: except Exception as e:
raise RuntimeError(f"Failed to load SAM2 model: {e}") raise RuntimeError(f"Failed to load SAM2 model: {e}")