Merge pull request #10339 from catboxanon/bf16 (d70c3a80) · Commits · github_fork / Stable Diffusion Webui

modules/safe.py

+1 −1

Original line number	Diff line number	Diff line
		@@ -40,7 +40,7 @@ class RestrictedUnpickler(pickle.Unpickler):
		return getattr(collections, name)
		if module == 'torch._utils' and name in ['_rebuild_tensor_v2', '_rebuild_parameter', '_rebuild_device_tensor_from_numpy']:
		return getattr(torch._utils, name)
		if module == 'torch' and name in ['FloatStorage', 'HalfStorage', 'IntStorage', 'LongStorage', 'DoubleStorage', 'ByteStorage', 'float32']:
		if module == 'torch' and name in ['FloatStorage', 'HalfStorage', 'IntStorage', 'LongStorage', 'DoubleStorage', 'ByteStorage', 'float32', 'BFloat16Storage']:
		return getattr(torch, name)
		if module == 'torch.nn.modules.container' and name in ['ParameterDict']:
		return getattr(torch.nn.modules.container, name)