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

fix test_names LoraManager test

Browse source
This commit is contained in:
limiteinductive 2024-01-23 12:52:15 +01:00 committed by Benjamin Trom
parent 421da6a3b6
commit 3b458f0d8d
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
tests/foundationals/latent_diffusion/test_lora_manager.py
View file

@ -75,7 +75,7 @@ def test_names(manager: SDLoraManager, weights: dict[str, torch.Tensor]) -> None
assert manager.names == ["pokemon-lora"] assert manager.names == ["pokemon-lora"]
manager.add_loras("pokemon-lora2", tensors=weights) manager.add_loras("pokemon-lora2", tensors=weights)
assert manager.names == ["pokemon-lora", "pokemon-lora2"] assert set(manager.names) == set(["pokemon-lora", "pokemon-lora2"])
def test_scales(manager: SDLoraManager, weights: dict[str, torch.Tensor]) -> None: def test_scales(manager: SDLoraManager, weights: dict[str, torch.Tensor]) -> None: