Issue with Multi-GPU Distributed Computing Using UCE Model

snap-stanford / UCE

UCE is a zero-shot foundation model for single-cell gene expression data

MIT License

120 stars 15 forks source link

Issue with Multi-GPU Distributed Computing Using UCE Model #37

Open kelimeike opened 1 month ago

kelimeike commented 1 month ago

Hi there,

I encountered a problem while calculating embeddings with the UCE model and setting args.multi_gpu=True. I received an error: AttributeError: 'TransformerModel' object has no attribute 'module'. Could you please help me understand what might be causing this and how to resolve it?

Thank you!

Yanay1 commented 1 month ago

Hi, are you launching the command with hugging face accelerator?