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"]
|
||||
|
||||
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:
|
||||
|
|
Loading…
Reference in a new issue