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