Issue with Multi-GPU Distributed Computing Using UCE Model

Question

Issue with Multi-GPU Distributed Computing Using UCE Model

Closed this issue 9 months ago · 1 comments

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!

Answer 1 · 2024-05-14T02:06:51.000Z

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