INNER CODE UNIT · Python

models

fofr/cog-face-to-sticker · helpers/ComfyUI_BRIA_AI_RMBG.py:7

    def models():
        return MODELS

    @staticmethod
    def add_weights(weights_to_download, node):
        if "class_type" in node and node["class_type"] in [
            "BRIA_RMBG_ModelLoader_Zho",
        ]:
            weights_to_download.extend(MODELS)

    @staticmethod
    def weights_map(base_url):
        return {
            model: {
                "url": f"{base_url}/custom_nodes/ComfyUI-BRIA_AI-RMBG/{model}.tar",
                "dest": "ComfyUI/custom_nodes/ComfyUI-BRIA_AI-RMBG/",
            }
            for model in MODELS

View source record →