Move test util to library

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

View File

View File

@@ -1,7 +1,7 @@
import torch
import pytest
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():

View File

@@ -2,7 +2,7 @@ import pytest
import torch
import torch.nn as nn
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