Skip to content

Commit 339a8d2

Browse files
committed
Implemented multi-GPU and batch llm_list.txt support
1 parent 0ef1ae4 commit 339a8d2

File tree

5 files changed

+1060
-4
lines changed

5 files changed

+1060
-4
lines changed

src/reptrace/__init__.py

Lines changed: 16 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,20 +5,34 @@
55
"DNAExtractionConfig",
66
"DNAExtractionResult",
77
"calc_dna",
8+
"calc_dna_parallel",
89
"calc_dna_batch",
910
]
1011

1112
__version__ = "0.1.0"
1213

1314

1415
def __getattr__(name: str):
15-
if name in {"DNAExtractionConfig", "DNAExtractionResult", "calc_dna", "calc_dna_batch"}:
16-
from .api import DNAExtractionConfig, DNAExtractionResult, calc_dna, calc_dna_batch
16+
if name in {
17+
"DNAExtractionConfig",
18+
"DNAExtractionResult",
19+
"calc_dna",
20+
"calc_dna_parallel",
21+
"calc_dna_batch",
22+
}:
23+
from .api import (
24+
DNAExtractionConfig,
25+
DNAExtractionResult,
26+
calc_dna,
27+
calc_dna_parallel,
28+
calc_dna_batch,
29+
)
1730

1831
exports = {
1932
"DNAExtractionConfig": DNAExtractionConfig,
2033
"DNAExtractionResult": DNAExtractionResult,
2134
"calc_dna": calc_dna,
35+
"calc_dna_parallel": calc_dna_parallel,
2236
"calc_dna_batch": calc_dna_batch,
2337
}
2438
return exports[name]

0 commit comments

Comments
 (0)