Type of the return value is error.

[YushunXiang]

System Info

all system

Information

• The official example scripts
• My own modified scripts

Reproduction

safetensors/bindings/python/py_src/safetensors/torch.py

Lines 189 to 194 in 8814598

	def load_model(
	model: torch.nn.Module,
	filename: Union[str, os.PathLike],
	strict: bool = True,
	device: Union[str, int] = "cpu",
	) -> Tuple[List[str], List[str]]:

The type of the return value should be Tuple[Set[str], Set[str]] instead of Tuple[List[str], List[str]]

Expected behavior

The type of the return value is Tuple[Set[str], Set[str]]