mirror of
https://github.com/finegrain-ai/refiners.git
synced 2024-11-09 23:12:02 +00:00
multi-upscaler: specify map_location when loading negative embedding
This commit is contained in:
parent
af1b302e78
commit
88325c3bbc
|
@ -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."
|
||||||
|
|
Loading…
Reference in a new issue