From f7d3a2a708f9a5122d3807664117e93619947917 Mon Sep 17 00:00:00 2001 From: Aniket Maurya Date: Wed, 13 Dec 2023 11:30:51 +0000 Subject: [PATCH] fixes --- src/fastserve/__init__.py | 1 + src/fastserve/__main__.py | 2 +- src/fastserve/models/__main__.py | 6 +++--- 3 files changed, 5 insertions(+), 4 deletions(-) diff --git a/src/fastserve/__init__.py b/src/fastserve/__init__.py index bfc56ee..5e173bc 100644 --- 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 index 95a727b..7bab108 100644 --- 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 index 4cfad7b..2cc09be 100644 --- 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")