Skip to content

Commit

Permalink
Retry with single core in geneformer
Browse files Browse the repository at this point in the history
  • Loading branch information
lazappi committed Jan 21, 2025
1 parent f45bd88 commit a7d675e
Showing 1 changed file with 15 additions and 11 deletions.
26 changes: 15 additions & 11 deletions src/methods/geneformer/config.vsh.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -21,20 +21,24 @@ info:
method_types: [embedding]
variants:
geneformer_12L_95M_i4096:
model: "gf-12L-95M-i4096"
model: gf-12L-95M-i4096
geneformer_6L_30M_i2048:
model: "gf-6L-30M-i2048"
model: gf-6L-30M-i2048
geneformer_12L_30M_i2048:
model: "gf-12L-30M-i2048"
model: gf-12L-30M-i2048
geneformer_20L_95M_i4096:
model: "gf-20L-95M-i4096"
model: gf-20L-95M-i4096

arguments:
- name: "--model"
type: "string"
- name: --model
type: string
description: String representing the Geneformer model to use
choices: ["gf-6L-30M-i2048", "gf-12L-30M-i2048", "gf-12L-95M-i4096", "gf-20L-95M-i4096"]
default: "gf-12L-95M-i4096"
choices:
- gf-6L-30M-i2048
- gf-12L-30M-i2048
- gf-12L-95M-i4096
- gf-20L-95M-i4096
default: gf-12L-95M-i4096

resources:
- type: python_script
Expand All @@ -48,9 +52,9 @@ engines:
setup:
- type: python
pip:
- pyarrow<15.0.0a0,>=14.0.1
- huggingface_hub
- git+https://huggingface.co/ctheodoris/Geneformer.git
- pyarrow<15.0.0a0,>=14.0.1
- huggingface_hub
- git+https://huggingface.co/ctheodoris/Geneformer.git

runners:
- type: executable
Expand Down

0 comments on commit a7d675e

Please sign in to comment.