Your current code snippet doesn’t show the usage of the model, but based on the error message it seems your are passing an Int into the model via:
data = 1
output = model(data)

Could you check the dtype of the inputs and make sure they are not overwritten by other variables with the same name?