Size mismatch in lr_find

I was following Lecture 5 on part 1 where Jeremy made his own model o explain the matrix multiplications. Here I used a 128 by 128 images but I cant get through the lr_find() i dont have any idea why is this so.
Here are the SS: