Finegrain/refiners
1
0
Fork 0
mirror of https://github.com/finegrain-ai/refiners.git synced 2024-11-21 21:58:47 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

multi-upscaler: specify map_location when loading negative embedding

Browse source
This commit is contained in:
Laurent 2024-07-12 12:05:17 +00:00 committed by Cédric Deltheil
parent af1b302e78
commit 88325c3bbc
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
src/refiners/foundationals/latent_diffusion/stable_diffusion_1/multi_upscaler.py
View file

@ -112,7 +112,9 @@ class MultiUpscalerAbstract(MultiDiffusion[T], ABC):
if path is None: if path is None:
return "" return ""
embeddings: Tensor | dict[str, Any] = torch.load(path, weights_only=True) # type: ignore embeddings: torch.Tensor | dict[str, Any] = torch.load( # type: ignore
path, weights_only=True, map_location=self.device
)
if isinstance(embeddings, dict): if isinstance(embeddings, dict):
assert key is not None, "Key must be provided to access the negative embedding." assert key is not None, "Key must be provided to access the negative embedding."