[Bug]: ModuleNotFoundError: No module named 'models.gpt2' about olive HOT 5 CLOSED

@akarym-sl you have a same module name models in your folder so the Python cannot get the correct models module from onnxruntime. Can you please rename your models folder to something else like model_script?

from olive.

@xiaoyu-work yes, the error is gone when I do

from olive.

However, I need to also modify the model save file somehow to work with the new directory. Is there no better way to avoid this issue?

from olive.

After saving the model not as the whole object, but only the state dict, the problem is no more. While I understand where the issue is coming from, this is my first time encountering the conflict between my directories and ones of a package. Therefore, I assume that there is a better way to implement this which avoids such conflicts.

from olive.

the namespace conflicts comes from the Python module implementation.

from olive.