alegendaryfish commited on
Commit
5343ca4
·
verified ·
1 Parent(s): 53a2f0d

Rename public-facing CodonGPT strings to CodonTranslator

Browse files
Files changed (1) hide show
  1. sampling.py +3 -3
sampling.py CHANGED
@@ -1,6 +1,6 @@
1
  #!/usr/bin/env python
2
  """
3
- Sampling script for generating codon sequences from trained CodonGPT models.
4
  Inputs are prepared exactly like training:
5
  - Species conditioning via SpeciesEmbeddingStore (fixed-size [B,Ds] or variable-length [B,Ls,Ds])
6
  - Protein conditioning via raw AA strings (ESM-C tokenization happens inside the model)
@@ -22,11 +22,11 @@ logging.basicConfig(
22
  datefmt="%m/%d/%Y %H:%M:%S",
23
  level=logging.INFO,
24
  )
25
- logger = logging.getLogger("codongpt.sample")
26
 
27
 
28
  def parse_args():
29
- p = argparse.ArgumentParser(description="Sample codon sequences from CodonGPT model")
30
 
31
  # Model
32
  p.add_argument("--model_path", "--model_dir", dest="model_path", type=str, required=True,
 
1
  #!/usr/bin/env python
2
  """
3
+ Sampling script for generating codon sequences from trained CodonTranslator models.
4
  Inputs are prepared exactly like training:
5
  - Species conditioning via SpeciesEmbeddingStore (fixed-size [B,Ds] or variable-length [B,Ls,Ds])
6
  - Protein conditioning via raw AA strings (ESM-C tokenization happens inside the model)
 
22
  datefmt="%m/%d/%Y %H:%M:%S",
23
  level=logging.INFO,
24
  )
25
+ logger = logging.getLogger("codontranslator.sample")
26
 
27
 
28
  def parse_args():
29
+ p = argparse.ArgumentParser(description="Sample codon sequences from a CodonTranslator model")
30
 
31
  # Model
32
  p.add_argument("--model_path", "--model_dir", dest="model_path", type=str, required=True,