TypeError: forward() missing 2 required positional arguments: 'input' and 'attn_text'

[Agiroy4712]

hello~, i'm, try to run the demo.py but I got this error. And another problem is that is always use all the gpus even though i set the gpu_num.
thank for reply !!
Traceback (most recent call last): File "demo.py", line 137, in <module> demo(opt) File "demo.py", line 55, in demo predss = model(image, text_for_pred, is_train=False)[0] File "/home/liaohaiqing/anaconda3/envs/text/lib/python3.7/site-packages/torch/nn/modules/module.py", line 727, in _call_impl result = self.forward(*input, **kwargs) File "/home/liaohaiqing/anaconda3/envs/text/lib/python3.7/site-packages/torch/nn/parallel/data_parallel.py", line 161, in forward outputs = self.parallel_apply(replicas, inputs, kwargs) File "/home/liaohaiqing/anaconda3/envs/text/lib/python3.7/site-packages/torch/nn/parallel/data_parallel.py", line 171, in parallel_apply return parallel_apply(replicas, inputs, kwargs, self.device_ids[:len(replicas)]) File "/home/liaohaiqing/anaconda3/envs/text/lib/python3.7/site-packages/torch/nn/parallel/parallel_apply.py", line 86, in parallel_apply output.reraise() File "/home/liaohaiqing/anaconda3/envs/text/lib/python3.7/site-packages/torch/_utils.py", line 428, in reraise raise self.exc_type(msg) TypeError: Caught TypeError in replica 4 on device 4. Original Traceback (most recent call last): File "/home/liaohaiqing/anaconda3/envs/text/lib/python3.7/site-packages/torch/nn/parallel/parallel_apply.py", line 61, in _worker output = module(*input, **kwargs) File "/home/liaohaiqing/anaconda3/envs/text/lib/python3.7/site-packages/torch/nn/modules/module.py", line 727, in _call_impl result = self.forward(*input, **kwargs) TypeError: forward() missing 2 required positional arguments: 'input' and 'attn_text'