Skip to content

Commit a2d54e7

Browse files
committed
from pypef.utils.helpers import get_device
1 parent 6b768ee commit a2d54e7

File tree

2 files changed

+5
-3
lines changed

2 files changed

+5
-3
lines changed

pypef/llm/esm_lora_tune.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,8 @@
2727
hf_logging.set_verbosity_error()
2828
from transformers import EsmForMaskedLM, EsmTokenizer
2929

30+
from pypef.utils.helpers import get_device
31+
3032
logger = logging.getLogger('pypef.llm.esm_lora_tune')
3133

3234

scripts/CLI/run_cli_tests_win.ps1

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ function ExitOnExitCode { if ($LastExitCode) {
2525

2626
### if using downloaded/locally stored pypef .py files:
2727
##########################################################################################################################
28-
ECHO Y | conda env remove -n pypef #
28+
Write-Output Y | conda env remove -n pypef #
2929
conda create -n pypef python=3.12 -y #
3030
conda activate pypef #
3131
$path=Get-Location #
@@ -331,13 +331,13 @@ Write-Host
331331
pypef encode -i 37_ANEH_variants.csv -e dca -w Sequence_WT_ANEH.fasta --params ANEH_72.6.params --threads $threads
332332
ExitOnExitCode
333333
Write-Host
334-
mv 37_ANEH_variants_dca_encoded.csv 37_ANEH_variants_plmc_dca_encoded.csv
334+
Move-Item 37_ANEH_variants_dca_encoded.csv 37_ANEH_variants_plmc_dca_encoded.csv
335335
ExitOnExitCode
336336
Write-Host
337337
pypef encode -i 37_ANEH_variants.csv -e dca -w Sequence_WT_ANEH.fasta --params GREMLIN
338338
ExitOnExitCode
339339
Write-Host
340-
mv 37_ANEH_variants_dca_encoded.csv 37_ANEH_variants_gremlin_dca_encoded.csv
340+
Move-Item 37_ANEH_variants_dca_encoded.csv 37_ANEH_variants_gremlin_dca_encoded.csv
341341
ExitOnExitCode
342342
Write-Host
343343

0 commit comments

Comments
 (0)