search

evolutionaryscale / esm

Other
798 stars 81 forks source link

Incompatible dependencies in python 3.12 #9

Open atong01 opened 6 days ago

atong01 commented 6 days ago

Current readme example fails when installed on python 3.12 on torch.compile.

Will be broken on torch<2.4 see https://github.com/pytorch/pytorch/issues/120233

Might suggest pinning python version for now? Or at least noting somewhere in README.

See below for error:

protein = model.generate(protein, GenerationConfig(track="sequence", num_steps=8, temperature=0.7))

File "/home/esm3/esm/utils/structure/affine3d.py", line 14, in maybe_compile
    return torch.compile(func) if x.device.type == "cuda" else func
           ^^^^^^^^^^^^^^^^^^^
  File "/home/esm3/lib/python3.12/site-packages/torch/__init__.py", line 1868, in compile
    raise RuntimeError("Dynamo is not supported on Python 3.12+")
RuntimeError: Dynamo is not supported on Python 3.12+