pub_soft/esm
3
0
Fork 0
mirror of https://github.com/evolutionaryscale/esm.git synced 2026-06-04 09:04:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

pyright ignore flash_attn import because it is guarded (#274)

Browse Source
This commit is contained in:
Ishaan Mathur
2025-09-18 13:54:10 -04:00
committed by GitHub
parent cbaccc1693
commit 97eb26b2fe
3 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
.github/workflows/ci.yml vendored
View File

@@ -25,9 +25,9 @@ jobs:
uses: actions/checkout@v4
- name: Setup Environment
uses: prefix-dev/setup-pixi@v0.8.1
uses: prefix-dev/setup-pixi@v0.9.0
with:
pixi-version: v0.47.0
pixi-version: v0.54.0
cache: false
cache-write: ${{ github.event_name == 'push' && github.ref_name == 'main' }}
@@ -43,9 +43,9 @@ jobs:
uses: actions/checkout@v4
- name: Setup Environment
uses: prefix-dev/setup-pixi@v0.8.1
uses: prefix-dev/setup-pixi@v0.9.0
with:
pixi-version: v0.47.0
pixi-version: v0.54.0
cache: false
cache-write: ${{ github.event_name == 'push' && github.ref_name == 'main' }}

2
esm/layers/attention.py
View File

@@ -8,7 +8,7 @@ from torch import nn
from esm.layers.rotary import RotaryEmbedding, TritonRotaryEmbedding
try:
from flash_attn import flash_attn_varlen_qkvpacked_func
from flash_attn import flash_attn_varlen_qkvpacked_func # type: ignore
except (ImportError, RuntimeError):
flash_attn_varlen_qkvpacked_func = None

4
tests/oss_pytests/test_oss_client.py
View File

@@ -53,7 +53,7 @@ def test_oss_esm3_client():
def test_oss_esmc_client():
assert URL is not None
sequence = "MALWMRLLPLLALLALAVUUPDPAAA"
sequence = "MALWMRLLPLLALLALAVPDPAAA"
model = "esmc-300m-2024-12"
esmc_client = client(model=model, url=URL, token=API_TOKEN)
@@ -75,7 +75,7 @@ def test_oss_esmc_client():
def test_oss_sequence_structure_forge_inference_client():
assert URL is not None
sequence = "MALWMRLLPLLALLALAVUUPDPAAA"
sequence = "MALWMRLLPLLALLALAVPDPAAA"
model = "esm3-small-2024-03"
client = SequenceStructureForgeInferenceClient(
model=model, url=URL, token=API_TOKEN