Unable to make use of text-bison-32k model using python SDK

[shrikumar_d]

Hi folks,
I'm trying to make use of the text-bison-32k model using Python SDK. I have the appropriate credentials set using the credentials JSON and initialized vertex ai. I'm getting the following error -
ValueError: The model you are trying to instantiate does not support the launch stage: PUBLIC_PREVIEW
I'm able to make use of the model from the UI and command line though. Is there something that I'm missing out on?

[todd_sudduth · Accepted solution]

I ran into this today, and I was scratching my head for a bit.  Here is the solution.  Change your "import" to be preview.

from vertexai.preview.language_models import TextGenerationModel

[shrikumar_d]

Thanks, Todd. It worked. 
Weird though, since the code that Vertex provides in the console doesn't contain "preview" in "import".