rafacost

/

DreamOmni2-7.6B-GGUF

like 16

Thanks for the feedback — appreciate you testing the node!

ComfyUI already handles extra_model_paths.yaml natively on startup. Any folders defined there (for example under unet, loras, etc.) are automatically registered by folder_paths, so the node should already detect models in those locations without extra code changes.

If your setup isn’t picking them up, please share your extra_model_paths.yaml structure and the ComfyUI version so we can double-check. But in general, no modification to dreamomni2.py should be required.

I found in dreamomni2.py, unet_dir = os.path.join(folder_paths.models_dir, "unet") only reads the location of the current ComfyUI, while the paths configured in extra_model_paths.yaml need to be obtained through the dedicated methods of the folder_paths module. ComfyUI need to use folder_paths.get_folder_paths("unet") to retrieve all configured paths (including both default and additional paths).

Thanks for the detailed contribution — your changes helped confirm that extra_model_paths works properly.
We’ve now implemented this functionality in a simpler, more maintainable way as part of the v0.2.0 release, which also adds caching and less verbose logs.
I’m closing this PR to keep the codebase clean, but your effort and testing were very helpful. 🙏