Skip to content

Commit

Permalink
Hotfix endpoints
Browse files Browse the repository at this point in the history
  • Loading branch information
@liambai
liambai committed Jan 27, 2025
1 parent 76e674e commit 1389738
Show file tree
Hide file tree
Showing 5 changed files with 75 additions and 75 deletions.
40 changes: 20 additions & 20 deletions interprot/endpoints/sae_inference/Dockerfile
View file
Original file line number Diff line number Diff line change
Expand Up @@ -15,26 +15,26 @@ RUN wget https://dl.fbaipublicfiles.com/fair-esm/models/esm2_t33_650M_UR50D.pt

# Weights for esm2_plm1280_l24_sae4096_100Kseqs.pt
RUN gdown https://drive.google.com/uc?id=1LtDUfcWQEwPuTdd127HEb_A2jQdRyKrU
# Weights for esm2_plm1280_l24_sae4096_k128_auxk512_antibody_seqs.ckpt
RUN gdown https://drive.google.com/uc?id=19aCVCVLleTc4QSiXZsi5hPqrE21duk6q
# Weights for esm2_plm1280_l4_sae4096_k64.ckpt
RUN gdown https://drive.google.com/uc?id=1yrfhQ4Qtcpe2v9oeiBl4csklcnGbamNp
# Weights for esm2_plm1280_l8_sae4096_k64_auxk640.ckpt
RUN gdown https://drive.google.com/uc?id=1m30OvYHZmtdI8l6F1GsWr8TnZr_0Q_ax
# Weights for esm2_plm1280_l12_sae4096_k64.ckpt
RUN gdown https://drive.google.com/uc?id=1UA9Y6EV9cgY-HtNjz9n46DE4nUCJ8S1U
# Weights for esm2_plm1280_l16_sae4096_k64_auxk640.ckpt
RUN gdown https://drive.google.com/uc?id=1f_kHYqrV9qw-RKgQUBX-p5hDntwSkANd
# Weights for esm2_plm1280_l20_sae4096_k64.ckpt
RUN gdown https://drive.google.com/uc?id=1W_2sU3V4zTw0crG0fduKNdJpkk7CXgsd
# Weights for esm2_plm1280_l24_sae4096_k64_auxk640.ckpt
RUN gdown https://drive.google.com/uc?id=1QfcQLWBH5t2Bt975bbRNS33fUPGpaFJN
# Weights for esm2_plm1280_l28_sae4096_k64.ckpt
RUN gdown https://drive.google.com/uc?id=1wvyl0yb4kGbnlMYQsJpl7JSmNDLnoNpu
# Weights for esm2_plm1280_l32_sae4096_k64.ckpt
RUN gdown https://drive.google.com/uc?id=1LXwEnDsgLpyCILyTrQv_W2yTLwCV-6IP
# Weights for esm2_plm1280_l33_sae4096_aux640.ckpt
RUN gdown https://drive.google.com/uc?id=1Ly7IQjAp3UcPOgQLCgV6BQiwknV32VZU
# # Weights for esm2_plm1280_l24_sae4096_k128_auxk512_antibody_seqs.ckpt
# RUN gdown https://drive.google.com/uc?id=19aCVCVLleTc4QSiXZsi5hPqrE21duk6q
# # Weights for esm2_plm1280_l4_sae4096_k64.ckpt
# RUN gdown https://drive.google.com/uc?id=1yrfhQ4Qtcpe2v9oeiBl4csklcnGbamNp
# # Weights for esm2_plm1280_l8_sae4096_k64_auxk640.ckpt
# RUN gdown https://drive.google.com/uc?id=1m30OvYHZmtdI8l6F1GsWr8TnZr_0Q_ax
# # Weights for esm2_plm1280_l12_sae4096_k64.ckpt
# RUN gdown https://drive.google.com/uc?id=1UA9Y6EV9cgY-HtNjz9n46DE4nUCJ8S1U
# # Weights for esm2_plm1280_l16_sae4096_k64_auxk640.ckpt
# RUN gdown https://drive.google.com/uc?id=1f_kHYqrV9qw-RKgQUBX-p5hDntwSkANd
# # Weights for esm2_plm1280_l20_sae4096_k64.ckpt
# RUN gdown https://drive.google.com/uc?id=1W_2sU3V4zTw0crG0fduKNdJpkk7CXgsd
# # Weights for esm2_plm1280_l24_sae4096_k64_auxk640.ckpt
# RUN gdown https://drive.google.com/uc?id=1QfcQLWBH5t2Bt975bbRNS33fUPGpaFJN
# # Weights for esm2_plm1280_l28_sae4096_k64.ckpt
# RUN gdown https://drive.google.com/uc?id=1wvyl0yb4kGbnlMYQsJpl7JSmNDLnoNpu
# # Weights for esm2_plm1280_l32_sae4096_k64.ckpt
# RUN gdown https://drive.google.com/uc?id=1LXwEnDsgLpyCILyTrQv_W2yTLwCV-6IP
# # Weights for esm2_plm1280_l33_sae4096_aux640.ckpt
# RUN gdown https://drive.google.com/uc?id=1Ly7IQjAp3UcPOgQLCgV6BQiwknV32VZU


WORKDIR /
Expand Down
20 changes: 10 additions & 10 deletions interprot/endpoints/sae_inference/handler.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -24,16 +24,16 @@
WEIGHTS_DIR = "/weights"
SAE_NAME_TO_CHECKPOINT = {
"SAE4096-L24": "esm2_plm1280_l24_sae4096_100Kseqs.pt",
"SAE4096-L24-ab": "esm2_plm1280_l24_sae4096_k128_auxk512_antibody_seqs.ckpt",
"SAE4096-L4": "esm2_plm1280_l4_sae4096_k64.ckpt",
"SAE4096-L8": "esm2_plm1280_l8_sae4096_k64_auxk640.ckpt",
"SAE4096-L12": "esm2_plm1280_l12_sae4096_k64.ckpt",
"SAE4096-L16": "esm2_plm1280_l16_sae4096_k64_auxk640.ckpt",
"SAE4096-L20": "esm2_plm1280_l20_sae4096_k64.ckpt",
"SAE4096-L24-v2": "esm2_plm1280_l24_sae4096_k64_auxk640.ckpt",
"SAE4096-L28": "esm2_plm1280_l28_sae4096_k64.ckpt",
"SAE4096-L32": "esm2_plm1280_l32_sae4096_k64.ckpt",
"SAE4096-L33": "esm2_plm1280_l33_sae4096_aux640.ckpt",
# "SAE4096-L24-ab": "esm2_plm1280_l24_sae4096_k128_auxk512_antibody_seqs.ckpt",
# "SAE4096-L4": "esm2_plm1280_l4_sae4096_k64.ckpt",
# "SAE4096-L8": "esm2_plm1280_l8_sae4096_k64_auxk640.ckpt",
# "SAE4096-L12": "esm2_plm1280_l12_sae4096_k64.ckpt",
# "SAE4096-L16": "esm2_plm1280_l16_sae4096_k64_auxk640.ckpt",
# "SAE4096-L20": "esm2_plm1280_l20_sae4096_k64.ckpt",
# "SAE4096-L24-v2": "esm2_plm1280_l24_sae4096_k64_auxk640.ckpt",
# "SAE4096-L28": "esm2_plm1280_l28_sae4096_k64.ckpt",
# "SAE4096-L32": "esm2_plm1280_l32_sae4096_k64.ckpt",
# "SAE4096-L33": "esm2_plm1280_l33_sae4096_aux640.ckpt",
}


Expand Down
40 changes: 20 additions & 20 deletions interprot/endpoints/steer_feature/Dockerfile
View file
Original file line number Diff line number Diff line change
Expand Up @@ -17,26 +17,26 @@ RUN wget https://dl.fbaipublicfiles.com/fair-esm/models/esm2_t33_650M_UR50D.pt

# Weights for esm2_plm1280_l24_sae4096_100Kseqs.pt
RUN gdown https://drive.google.com/uc?id=1LtDUfcWQEwPuTdd127HEb_A2jQdRyKrU
# Weights for esm2_plm1280_l24_sae4096_k128_auxk512_antibody_seqs.ckpt
RUN gdown https://drive.google.com/uc?id=19aCVCVLleTc4QSiXZsi5hPqrE21duk6q
# Weights for esm2_plm1280_l4_sae4096_k64.ckpt
RUN gdown https://drive.google.com/uc?id=1yrfhQ4Qtcpe2v9oeiBl4csklcnGbamNp
# Weights for esm2_plm1280_l8_sae4096_k64_auxk640.ckpt
RUN gdown https://drive.google.com/uc?id=1m30OvYHZmtdI8l6F1GsWr8TnZr_0Q_ax
# Weights for esm2_plm1280_l12_sae4096_k64.ckpt
RUN gdown https://drive.google.com/uc?id=1UA9Y6EV9cgY-HtNjz9n46DE4nUCJ8S1U
# Weights for esm2_plm1280_l16_sae4096_k64_auxk640.ckpt
RUN gdown https://drive.google.com/uc?id=1f_kHYqrV9qw-RKgQUBX-p5hDntwSkANd
# Weights for esm2_plm1280_l20_sae4096_k64.ckpt
RUN gdown https://drive.google.com/uc?id=1W_2sU3V4zTw0crG0fduKNdJpkk7CXgsd
# Weights for esm2_plm1280_l24_sae4096_k64_auxk640.ckpt
RUN gdown https://drive.google.com/uc?id=1QfcQLWBH5t2Bt975bbRNS33fUPGpaFJN
# Weights for esm2_plm1280_l28_sae4096_k64.ckpt
RUN gdown https://drive.google.com/uc?id=1wvyl0yb4kGbnlMYQsJpl7JSmNDLnoNpu
# Weights for esm2_plm1280_l32_sae4096_k64.ckpt
RUN gdown https://drive.google.com/uc?id=1LXwEnDsgLpyCILyTrQv_W2yTLwCV-6IP
# Weights for esm2_plm1280_l33_sae4096_aux640.ckpt
RUN gdown https://drive.google.com/uc?id=1Ly7IQjAp3UcPOgQLCgV6BQiwknV32VZU
# # Weights for esm2_plm1280_l24_sae4096_k128_auxk512_antibody_seqs.ckpt
# RUN gdown https://drive.google.com/uc?id=19aCVCVLleTc4QSiXZsi5hPqrE21duk6q
# # Weights for esm2_plm1280_l4_sae4096_k64.ckpt
# RUN gdown https://drive.google.com/uc?id=1yrfhQ4Qtcpe2v9oeiBl4csklcnGbamNp
# # Weights for esm2_plm1280_l8_sae4096_k64_auxk640.ckpt
# RUN gdown https://drive.google.com/uc?id=1m30OvYHZmtdI8l6F1GsWr8TnZr_0Q_ax
# # Weights for esm2_plm1280_l12_sae4096_k64.ckpt
# RUN gdown https://drive.google.com/uc?id=1UA9Y6EV9cgY-HtNjz9n46DE4nUCJ8S1U
# # Weights for esm2_plm1280_l16_sae4096_k64_auxk640.ckpt
# RUN gdown https://drive.google.com/uc?id=1f_kHYqrV9qw-RKgQUBX-p5hDntwSkANd
# # Weights for esm2_plm1280_l20_sae4096_k64.ckpt
# RUN gdown https://drive.google.com/uc?id=1W_2sU3V4zTw0crG0fduKNdJpkk7CXgsd
# # Weights for esm2_plm1280_l24_sae4096_k64_auxk640.ckpt
# RUN gdown https://drive.google.com/uc?id=1QfcQLWBH5t2Bt975bbRNS33fUPGpaFJN
# # Weights for esm2_plm1280_l28_sae4096_k64.ckpt
# RUN gdown https://drive.google.com/uc?id=1wvyl0yb4kGbnlMYQsJpl7JSmNDLnoNpu
# # Weights for esm2_plm1280_l32_sae4096_k64.ckpt
# RUN gdown https://drive.google.com/uc?id=1LXwEnDsgLpyCILyTrQv_W2yTLwCV-6IP
# # Weights for esm2_plm1280_l33_sae4096_aux640.ckpt
# RUN gdown https://drive.google.com/uc?id=1Ly7IQjAp3UcPOgQLCgV6BQiwknV32VZU

WORKDIR /

Expand Down
20 changes: 10 additions & 10 deletions interprot/endpoints/steer_feature/handler.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -24,16 +24,16 @@
WEIGHTS_DIR = "/weights"
SAE_NAME_TO_CHECKPOINT = {
"SAE4096-L24": "esm2_plm1280_l24_sae4096_100Kseqs.pt",
"SAE4096-L24-ab": "esm2_plm1280_l24_sae4096_k128_auxk512_antibody_seqs.ckpt",
"SAE4096-L4": "esm2_plm1280_l4_sae4096_k64.ckpt",
"SAE4096-L8": "esm2_plm1280_l8_sae4096_k64_auxk640.ckpt",
"SAE4096-L12": "esm2_plm1280_l12_sae4096_k64.ckpt",
"SAE4096-L16": "esm2_plm1280_l16_sae4096_k64_auxk640.ckpt",
"SAE4096-L20": "esm2_plm1280_l20_sae4096_k64.ckpt",
"SAE4096-L24-v2": "esm2_plm1280_l24_sae4096_k64_auxk640.ckpt",
"SAE4096-L28": "esm2_plm1280_l28_sae4096_k64.ckpt",
"SAE4096-L32": "esm2_plm1280_l32_sae4096_k64.ckpt",
"SAE4096-L33": "esm2_plm1280_l33_sae4096_aux640.ckpt",
# "SAE4096-L24-ab": "esm2_plm1280_l24_sae4096_k128_auxk512_antibody_seqs.ckpt",
# "SAE4096-L4": "esm2_plm1280_l4_sae4096_k64.ckpt",
# "SAE4096-L8": "esm2_plm1280_l8_sae4096_k64_auxk640.ckpt",
# "SAE4096-L12": "esm2_plm1280_l12_sae4096_k64.ckpt",
# "SAE4096-L16": "esm2_plm1280_l16_sae4096_k64_auxk640.ckpt",
# "SAE4096-L20": "esm2_plm1280_l20_sae4096_k64.ckpt",
# "SAE4096-L24-v2": "esm2_plm1280_l24_sae4096_k64_auxk640.ckpt",
# "SAE4096-L28": "esm2_plm1280_l28_sae4096_k64.ckpt",
# "SAE4096-L32": "esm2_plm1280_l32_sae4096_k64.ckpt",
# "SAE4096-L33": "esm2_plm1280_l33_sae4096_aux640.ckpt",
}


Expand Down
30 changes: 15 additions & 15 deletions viz/src/SAEConfigs.ts
View file
Original file line number Diff line number Diff line change
Expand Up @@ -420,7 +420,7 @@ export const SAE_CONFIGS: Record<string, SAEConfig> = {
numHiddenDims: 4096,
plmLayer: 4,
defaultDim: 0,
supportsCustomSequence: true,
supportsCustomSequence: false,
curated: [],
},
"SAE4096-L8": {
Expand All @@ -429,7 +429,7 @@ export const SAE_CONFIGS: Record<string, SAEConfig> = {
numHiddenDims: 4096,
plmLayer: 8,
defaultDim: 0,
supportsCustomSequence: true,
supportsCustomSequence: false,
curated: [],
},
"SAE4096-L12": {
Expand All @@ -438,7 +438,7 @@ export const SAE_CONFIGS: Record<string, SAEConfig> = {
numHiddenDims: 4096,
plmLayer: 12,
defaultDim: 0,
supportsCustomSequence: true,
supportsCustomSequence: false,
curated: [],
},
"SAE4096-L16": {
Expand All @@ -447,7 +447,7 @@ export const SAE_CONFIGS: Record<string, SAEConfig> = {
numHiddenDims: 4096,
plmLayer: 16,
defaultDim: 0,
supportsCustomSequence: true,
supportsCustomSequence: false,
curated: [],
},
"SAE4096-L20": {
Expand All @@ -456,7 +456,7 @@ export const SAE_CONFIGS: Record<string, SAEConfig> = {
numHiddenDims: 4096,
plmLayer: 20,
defaultDim: 0,
supportsCustomSequence: true,
supportsCustomSequence: false,
curated: [],
},
"SAE4096-L24-v2": {
Expand All @@ -466,7 +466,7 @@ export const SAE_CONFIGS: Record<string, SAEConfig> = {
numHiddenDims: 4096,
plmLayer: 24,
defaultDim: 0,
supportsCustomSequence: true,
supportsCustomSequence: false,
curated: [],
},
"SAE4096-L28": {
Expand All @@ -475,7 +475,7 @@ export const SAE_CONFIGS: Record<string, SAEConfig> = {
numHiddenDims: 4096,
plmLayer: 28,
defaultDim: 0,
supportsCustomSequence: true,
supportsCustomSequence: false,
curated: [],
},
"SAE4096-L32": {
Expand All @@ -484,7 +484,7 @@ export const SAE_CONFIGS: Record<string, SAEConfig> = {
numHiddenDims: 4096,
plmLayer: 32,
defaultDim: 0,
supportsCustomSequence: true,
supportsCustomSequence: false,
curated: [],
},
"SAE4096-L33": {
Expand All @@ -493,7 +493,7 @@ export const SAE_CONFIGS: Record<string, SAEConfig> = {
numHiddenDims: 4096,
plmLayer: 33,
defaultDim: 0,
supportsCustomSequence: true,
supportsCustomSequence: false,
curated: [],
},
"SAE4096-L24-ab": {
Expand All @@ -512,7 +512,7 @@ export const SAE_CONFIGS: Record<string, SAEConfig> = {
"QSALTQPASVSGSPGQSITISCTGTSSDVGGYNYVSWYQQHPGKAPKLMIYDVSKRPSGVSNRFSGSKSGNTASLTISGLQSEDEADYYCNSLTSISTWVFGGGTKLTVLGQPKAAPSVTLFPPSSEELQANKATLVCLISDFYPGAVTVAWKADSSPVKAGVETTTPSKQSNNKYAASSYLSLTPEQWKSHRSYSCQVTHEGSTVEKTVAPTECS",
},
defaultDim: 2699,
supportsCustomSequence: true,
supportsCustomSequence: false,
curated: [
{
name: "CDR2",
Expand Down Expand Up @@ -666,7 +666,7 @@ export const SAE_CONFIGS: Record<string, SAEConfig> = {
numHiddenDims: 8192,
plmLayer: 24,
defaultDim: 0,
supportsCustomSequence: true,
supportsCustomSequence: false,
curated: [],
},
"SAE8192-L24-K32": {
Expand All @@ -675,7 +675,7 @@ export const SAE_CONFIGS: Record<string, SAEConfig> = {
numHiddenDims: 8192,
plmLayer: 24,
defaultDim: 0,
supportsCustomSequence: true,
supportsCustomSequence: false,
curated: [],
},
"SAE8192-L24-K64": {
Expand All @@ -684,7 +684,7 @@ export const SAE_CONFIGS: Record<string, SAEConfig> = {
numHiddenDims: 8192,
plmLayer: 24,
defaultDim: 0,
supportsCustomSequence: true,
supportsCustomSequence: false,
curated: [],
},
"SAE8192-L24-K128": {
Expand All @@ -693,7 +693,7 @@ export const SAE_CONFIGS: Record<string, SAEConfig> = {
numHiddenDims: 8192,
plmLayer: 24,
defaultDim: 0,
supportsCustomSequence: true,
supportsCustomSequence: false,
curated: [],
},
"SAE8192-L24-K256": {
Expand All @@ -702,7 +702,7 @@ export const SAE_CONFIGS: Record<string, SAEConfig> = {
numHiddenDims: 8192,
plmLayer: 24,
defaultDim: 0,
supportsCustomSequence: true,
supportsCustomSequence: false,
curated: [],
},
};

0 comments on commit 1389738

Please sign in to comment.