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