AttributeError: module 't5x.partitioning' has no attribute 'ModelBasedPjitPartitioner'

#1
by deleted - opened

In https://github.com/google-research/t5x/blob/main/t5x/partitioning.py
I see PjitPartitioner and BasePjitPartitioner but no ModelBasedPjitPartitioner

deleted changed discussion title from fix it to AttributeError: module 't5x.partitioning' has no attribute 'ModelBasedPjitPartitioner'
deleted

confirmed PjitPartitioner is the correct replacement, see https://huggingface.co/spaces/vilimus/mt3-test

deleted changed discussion status to closed

Sign up or log in to comment