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

Before Change

    shared_module_key = getattr(module_config, "shared_module_key", None)
    typed_shared_module_key = (shared_module_key, type(module_config))
    load_path = getattr(module_config, "load_path", None)
    is_torchscript_load_path = load_path and zipfile.is_zipfile(
        PathManager.get_local_path(load_path)
    )
    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