thomas0809 / MolScribe

Robust Molecular Structure Recognition with Image-to-Graph Generation
MIT License
153 stars 31 forks source link

Update setup.py #5

Closed hanzlika closed 1 year ago

hanzlika commented 1 year ago

Remove the upper torch dependency restriction.

As of now, one of the dependecies timm 0.4.12 requires torchvision. Both available torchvision version (0.15.1 and 0.15.2) require torch above 2.0

This creates a dependency clash where molscribe requires torch<2.0 and torchivision requires torch==2.0.1

The current version of Molscribe works with torch==2.0.1