mirror of
https://github.com/finegrain-ai/refiners.git
synced 2024-11-09 23:12:02 +00:00
fix test_names LoraManager test
This commit is contained in:
parent
421da6a3b6
commit
3b458f0d8d
|
@ -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:
|
||||||
|
|
Loading…
Reference in a new issue