17 Commits
Author SHA1 Message Date
Mara Broda 802e8e9688 add accelerated apple silicon support and cpu fallback 2025-01-05 08:40:21 +01:00
Jan Minar 3ae2228e89 Offload models in low-VRAM environments
Performance should now match https://github.com/kijai/ComfyUI-Florence2
2024-10-23 21:04:19 +00:00
Jan Minar 487b0510ab fixes (hopefully): torch.OutOfMemoryError: Allocation on device
- Fix the SAM2 model being loaded on every run due to string being compared to
  class instance
- Manually unload the old SAM2 model from VRAM before loading the new one
- For a good measure, garbage-collect the RAM version as well
2024-10-09 20:17:31 +01:00
Jan Minar 7b191f63cf allow to select device
this should fix the problem of not being able to run the node on Windows
-- just select device: "cpu" in the dropdown
2024-10-04 01:27:36 +01:00
Jan Minar e5dffea0c6 Make good
* add masked image output
* label node outputs
* lazy load models
* add SAM model selector to the node
2024-09-30 02:32:52 +00:00
Jan Minar 459e9fa9a6 works in ComfyUI
note that this breaks the command-line interface
2024-09-30 01:30:41 +00:00
Jan Minar 014a3ad88d working from the commandline
Various changes made to make it work from the command line

- Most of the functionality has been commented out
    - Gradio-specific code
    - HTML UI
    - Video processing
    - Operation modes other than mask generation
2024-09-29 20:39:16 +00:00
SkalskiP c803a91309 Florence-2 + SAM2 video demo fix 2024-08-06 00:45:35 +02:00
SkalskiP 4a3364a0c3 add mp4 to lfs 2024-08-06 00:39:11 +02:00
SkalskiP 488d99ebc8 ready to migrate to ZERO 2024-08-06 00:11:21 +02:00
SkalskiP ae5c0ef4dd markdown improvements 2024-08-05 17:10:49 +02:00
SkalskiP 5ae5bcae46 use smaller model to improve CUDA usage 2024-08-01 14:58:49 +02:00
SkalskiP 576e22aaff open vocabulary detection with Florence2 + masks with SAM2 2024-08-01 14:42:04 +02:00
SkalskiP d1212b2598 update device to CUDA 2024-08-01 13:08:20 +02:00
SkalskiP 57700a8675 label text color updated 2024-08-01 13:01:38 +02:00
SkalskiP 2fbf361078 SAM2 added 2024-08-01 12:47:29 +02:00
SkalskiP baea9b2286 initial commit with Florence-2 2024-08-01 10:15:32 +02:00