LLava: deepspeed не может обнаружить редактируемый установленный пакет/модуль Python

Question

Я установил пакет (модель llava из github) какpython install -e .

В моей среде conda я загружаю llava как:

      >>python
>>import llava

Я поместил импорт в файл .py, когда использовал "python main.py. я могу импортироватьllava. Но используяdeepspeedв файле оболочки как:

      deepspeed main.py --args
>> sh run.sh

Это дало мне ошибку:

      ModuleNotFoundError: No module named 'llava'

Обратите внимание, что одна и та же конфигурация работает в одной системе, но не работает в другой.