Assertion error [5242880, 1] != [4096, 2560] when trying to deploy into Sagemaker Endpoint

#1
by LorenzoCevolaniAXA - opened

I am trying to deploy this model into a sagemaker endpoint and I have run into the following error:
AssertionError: [5242880, 1] != [4096, 2560]

I think this is due to some reshaping of the matrices inside the model.
What would be a fix on my side? or on your side

Sign up or log in to comment