fc52f79ed3dda250875b3ce9f3c241b80d643ef3,pytext/models/module.py,,create_module,#Any#,23

Before Change

    module = SHARED_MODULE_REGISTRY.get(typed_shared_module_key)
    if not module:
        if is_torchscript_load_path:
            with PathManager.open(load_path, "rb") as load_file:
                module = torch.jit.load(load_file)
        else:
            module = create_fn(module_config, *args, **kwargs)

    name = type(module).__name__

After Change

            with PathManager.open(load_path, "rb") as load_file:
                loaded_module = torch.load(load_file, map_location="cpu")

            if isinstance(loaded_module, dict):
                // Loaded module is a state dict
                module = create_fn(module_config, *args, **kwargs)
                module.load_state_dict(loaded_module)
            else:
                // Loaded module is a torchscripted module
                module = loaded_module

            name = type(module).__name__
            print(f"Loaded state of module {name} from {load_path} ...")

        else:

Instances