VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-18 01:30:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

Move test util to library

Browse Source
This commit is contained in:
rockerBOO
2025-03-25 18:41:58 -04:00
parent adb77d4196
commit c5c07a40c5
4 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

0
tests/test_util.py → library/test_util.py
View File

0
tests/__init__.py
View File

2
tests/library/test_lora_util.py
View File

@@ -1,7 +1,7 @@
import torch import torch
import pytest import pytest
from library.lora_util import initialize_pissa from library.lora_util import initialize_pissa
from ..test_util import generate_synthetic_weights from library.test_util import generate_synthetic_weights
def test_initialize_pissa_basic(): def test_initialize_pissa_basic():

2
tests/networks/test_lora_flux.py
View File

@@ -2,7 +2,7 @@ import pytest
import torch import torch
import torch.nn as nn import torch.nn as nn
from networks.lora_flux import LoRAModule, LoRANetwork, create_network from networks.lora_flux import LoRAModule, LoRANetwork, create_network
from ..test_util import generate_synthetic_weights from library.test_util import generate_synthetic_weights
from unittest.mock import MagicMock from unittest.mock import MagicMock