vertex AI - trained model with AutoML

[reshubisen]

I have trained a model and I'm able to fetch predictions from the endpoint. However, when I try to use the model in the following code:

pythonCopy code

model = TextGenerationModel.from_pretrained("*********")

I'm getting an error that says:

"NotFound: 404 Publisher Model publishers/google/models/******** is not found."

Could you please tell me how I can use my trained model in this code?

Is there any additional context or information you can provide to help with this issue?

vertexai.preview.language_models import TextGenerationModel

def write_a_function(temperature: float = 0.5) -> object:

"""Example of using Code Chat Model to write a function."""

parameters = {

"temperature": temperature, # Temperature controls the degree of randomness in token selection.

"max_output_tokens": 1024, # Token limit determines the maximum amount of text output.

}

model = TextGenerationModel.from_pretrained("*********")

response = model.predict(

'Give me all resource list',

**parameters,

)

print(f"Response from Model: {response }")