runtime error

thon3.10/site-packages/gradio/blocks.py", line 830, in call_function prediction = await anyio.to_thread.run_sync( File "/home/user/.local/lib/python3.10/site-packages/anyio/to_thread.py", line 33, in run_sync return await get_asynclib().run_sync_in_worker_thread( File "/home/user/.local/lib/python3.10/site-packages/anyio/_backends/_asyncio.py", line 877, in run_sync_in_worker_thread return await future File "/home/user/.local/lib/python3.10/site-packages/anyio/_backends/_asyncio.py", line 807, in run result = context.run(func, *args) File "/home/user/app/app.py", line 97, in igredent outputs = model.sample(image_tensor, greedy=True, File "/home/user/app/model.py", line 230, in sample ids, probs = self.recipe_decoder.sample(input_feats, input_mask, greedy, temperature, beam, img_features, 0, File "/home/user/app/modules/transformer_decoder.py", line 353, in sample outputs, _ = self.forward(ingr_features, ingr_mask, torch.stack(sampled_ids, 1), File "/home/user/app/modules/transformer_decoder.py", line 310, in forward x = layer( File "/home/user/.local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1518, in _wrapped_call_impl return self._call_impl(*args, **kwargs) File "/home/user/.local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1527, in _call_impl return forward_call(*args, **kwargs) File "/home/user/app/modules/transformer_decoder.py", line 208, in forward x, _ = self.cond_att(query=x, File "/home/user/.local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1518, in _wrapped_call_impl return self._call_impl(*args, **kwargs) File "/home/user/.local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1527, in _call_impl return forward_call(*args, **kwargs) File "/home/user/app/modules/multihead_attention.py", line 126, in forward attn_weights = attn_weights.float().masked_fill( RuntimeError: masked_fill_ only supports boolean masks, but got mask with dtype unsigned char

Container logs: