runtime error

Traceback (most recent call last): File "/home/user/app/app.py", line 18, in <module> pipe = pipe.to(device) AttributeError: 'Transformer2DModelOutput' object has no attribute 'to'

Container logs: