briaai

/

RMBG-2.0

like 1.1k

Follow

BRIA AI 13.5k

so, while running the default google colab notebook, i got this

from transformers import pipeline

pipe = pipeline("image-segmentation", model="briaai/RMBG-2.0", trust_remote_code=True)

RuntimeError                              Traceback (most recent call last)
/tmp/ipython-input-4702/1249673979.py in <cell line: 0>()
      2 from transformers import pipeline
      3 
----> 4 pipe = pipeline("image-segmentation", model="briaai/RMBG-2.0", trust_remote_code=True)

1 frames
/usr/local/lib/python3.12/dist-packages/transformers/pipelines/__init__.py in clean_custom_task(task_info)
    531 
    532     if "impl" not in task_info:
--> 533         raise RuntimeError("This model introduces a custom pipeline without specifying its implementation.")
    534     pt_class_names = task_info.get("pt", ())
    535     if isinstance(pt_class_names, str):

RuntimeError: This model introduces a custom pipeline without specifying its implementation.