amp
This commit is contained in:
@@ -50,7 +50,6 @@ class SAM2StreamingProcessor:
|
|||||||
vos_optimized=False, # Disable to avoid CUDA graph issues
|
vos_optimized=False, # Disable to avoid CUDA graph issues
|
||||||
hydra_overrides_extra=[
|
hydra_overrides_extra=[
|
||||||
"++model.compile_image_encoder=false", # Disable compilation
|
"++model.compile_image_encoder=false", # Disable compilation
|
||||||
"++model.memory_attention.use_amp=false", # Disable AMP for stability
|
|
||||||
]
|
]
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user