1
resposta

[Bug] Erro BriaRMGB

Insira aqui a descrição dessa imagem para ajudar na acessibilidade

1 resposta

Olá Matheus! Como vai?

Esse erro acontece durante o carregamento do modelo Briaai/RMBG-1.4 no pipeline do transformers. A mensagem:

'BriarMBG' object has no attribute 'all_tied_weights_keys'

indica que o código do modelo está tentando acessar um atributo interno que não existe na versão da biblioteca transformers instalada no ambiente.

Isso ocorre porque o modelo utiliza trust_remote_code=True, o que faz com que o transformers execute código Python do próprio repositório do modelo. Esse código foi desenvolvido para funcionar com uma versão específica da biblioteca, e quando usamos uma versão diferente pode surgir esse tipo de incompatibilidade.

Como resolver

A solução mais comum é ajustar a versão do transformers para uma versão compatível. Você pode tentar atualizar ou fixar uma versão estável da biblioteca:

pip install -U transformers

Caso o erro continue, tente instalar uma versão específica:

pip install transformers==4.37.2

Depois disso, reinicie o ambiente (notebook) e execute o código novamente.

O erro não está relacionado à imagem ou ao código do pipeline em si, mas sim a uma incompatibilidade entre o modelo carregado e a versão da biblioteca transformers utilizada no ambiente.

Se mesmo após ajustar a versão da biblioteca o problema persistir, vale verificar também as versões do torch e garantir que o ambiente foi reiniciado após a instalação.

Qualquer dúvida, me avise.

Abraços e bons estudos!