Skip to content

Commit f260ba2

Browse files
authored
Update sample_accessory.py
1 parent 8bd6339 commit f260ba2

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

sample_accessory.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,9 +22,9 @@
2222
from data import DataNoReportException, ItemProcessor, MyDataset, read_general
2323
from torchvision import transforms
2424

25-
import models_infilling as models
25+
import models_accessory as models
2626
from transport import Sampler, create_transport
27-
from models.lora import replace_linear_with_lora
27+
from models_accessory.lora import replace_linear_with_lora
2828

2929
#############################################################################
3030
# Condition Generator #

0 commit comments

Comments
 (0)