trim model ced1928
EmaadKhwaja commited on
How to use HuangLab/CELL-E_2_OpenCell_640 with Transformers:
# Load model directly
from transformers import AutoModel
model = AutoModel.from_pretrained("HuangLab/CELL-E_2_OpenCell_640", device_map="auto")