removed weak_module, weak_script_method

According to pytorch update: weak_module , weak_script_method no longer exist in torch._jit_internal. fix according to this discussion