torch2trt import error

[Senkai350]

TensorRT weight Generator will process the image with height 720 and width 1280 
Traceback (most recent call last):
  File "/Project/FAST_Anime_VSR/main.py", line 107, in <module>
    main()
  File "/Project/FAST_Anime_VSR/main.py", line 97, in main
    parallel_process(input_path, output_path, parallel_num=configuration.process_num)
  File "/Project/FAST_Anime_VSR/process/single_video.py", line 212, in parallel_process
    model_full_name, model_partition_name = weight_justify(configuration, input_path)
  File "/Project/FAST_Anime_VSR/process/single_video.py", line 92, in weight_justify
    generate_weight(h, w)               # It will automatically read model name we need
  File "/Project/FAST_Anime_VSR/tensorrt_weight_generator/weight_generator.py", line 257, in generate_weight
    tensorrt_transform_execute(lr_h, lr_width)
  File "/Project/FAST_Anime_VSR/tensorrt_weight_generator/weight_generator.py", line 206, in tensorrt_transform_execute
    ins.run()
  File "/Project/FAST_Anime_VSR/tensorrt_weight_generator/weight_generator.py", line 160, in run
    self.weight_generate()
  File "/Project/FAST_Anime_VSR/tensorrt_weight_generator/weight_generator.py", line 145, in weight_generate
    self.model_weight_transform(self.sample_input)
  File "/Project/FAST_Anime_VSR/tensorrt_weight_generator/weight_generator.py", line 92, in model_weight_transform
    from Real_ESRGAN.rrdb import RRDBNet
  File "/Project/FAST_Anime_VSR/Real_ESRGAN/rrdb.py", line 16, in <module>
    import Real_ESRGAN.torch2trt_fix    # For the bug override, don't delete it
  File "/Project/FAST_Anime_VSR/Real_ESRGAN/torch2trt_fix.py", line 4, in <module>
    from torch2trt.module_test import add_module_test
ModuleNotFoundError: No module named 'torch2trt.module_test'

i got this error on latest version