Exactly! they only get 2 feet tall max. so 2 gallons is plenty. its not recommended that they get planted into the ground either.
i just transplanted an Auto Jock Horror into a 2 gal on friday and thats where it will stay. its 16 days old and 6-7 inches tall but no preflowers yet.