Fix imports

[rasbt]

Fixes an import issue Aaron was having.

[awaelchli]

@rasbt It would be nice if you could paste the error you got into the PR description. If the import order matters, it is often a sign of circular imports. And if that's the case, we should try to remove the circular dependency.

[rasbt]

Good point @awaelchli . In this case it was not the circular dependency but issues with relative imports:

Traceback (most recent call last):
  File "/teamspace/studios/this_studio/server.py", line 15, in <module>
    from litgpt.generate.base import generate, next_token
  File "/teamspace/studios/this_studio/litgpt/litgpt/generate/base.py", line 17, in <module>
    from litgpt import GPT, Config, PromptStyle, Tokenizer
ImportError: cannot import name 'GPT' from 'litgpt' (unknown location)

[awaelchli]

But we don't have any relative imports LitGPT. Of course, the expectation is that the package is installed. Did the user not install the package? How could changing the import order fix something like that?

[rasbt]

I don't know exactly why it is happening, but it's not the order but changing

from litgpt import GPT

to

from litgpt.model import GPT

etc.

I remember having similar issues when implementing the API in #1463 even though I have LitGPT installed as a package. I don't fully understand why it's happening.

[rasbt]

Maybe @apaz-cli can provide more context here.