fixes

gradsflow · Dec 13, 2023 · f7d3a2a · f7d3a2a
1 parent d25cf9d
commit f7d3a2a
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git a/src/fastserve/__init__.py b/src/fastserve/__init__.py
@@ -1,3 +1,4 @@
 """A Python Project"""
+from .base_fastserve import FastServe
 
 __version__ = "0.0.1"
diff --git a/src/fastserve/__main__.py b/src/fastserve/__main__.py
@@ -1,4 +1,4 @@
-from .base_fastserve import FastServe
+from fastserve import FastServe
 
 serve = FastServe()
 serve.run_server()
diff --git a/src/fastserve/models/__main__.py b/src/fastserve/models/__main__.py
@@ -2,9 +2,9 @@
 
 from fastserve.utils import get_default_device
 
-from . import ServeSDXLTurbo
-from .llama_cpp import ServeLlamaCpp
-from .ssd import ServeSSD1B
+from fastserve.models.sdxl_turbo import ServeSDXLTurbo
+from fastserve.models.llama_cpp import ServeLlamaCpp
+from fastserve.models.ssd import ServeSSD1B
 
 parser = argparse.ArgumentParser(description="Serve models with FastServe")
 parser.add_argument("--model", type=str, required=True, help="Name of the model")