Fix bug where artifacts wouldn't deserialize onto CPU

terraytherapeutics / COATI

COATI: multi-modal contrastive pre-training for representing and traversing chemical space

Apache License 2.0

96 stars 5 forks source link

Closed ecwterray closed 6 months ago

ecwterray commented 6 months ago

Responding to Issue #4 , updated IO code to better support deserializing onto CPU.