update fn

README.md

@@ -24,19 +24,18 @@ python -m fastserve
 from fastserve.models import ServeLlamaCpp
 
 model_path = "openhermes-2-mistral-7b.Q5_K_M.gguf"
-serve = ServeLlamaCpp(model_path=model_path,)
+serve = ServeLlamaCpp(model_path=model_path, )
 serve.run_server()
 ```
 
 or, run `python -m fastserve.models --model llama-cpp --model_path openhermes-2-mistral-7b.Q5_K_M.gguf` from terminal.
 
-
 ### Serve SDXL Turbo
 
 ```python
-from fastserve.models import ServeSSD1B
+from fastserve.models import ServeSDXLTurbo
 
-serve = ServeSSD1B(device="cuda", batch_size=2, timeout=1)
+serve = ServeSDXLTurbo(device="cuda", batch_size=2, timeout=1)
 serve.run_server()
 ```
 

src/fastserve/models/__init__.py

@@ -1,3 +1,3 @@
 from .ssd import ServeSSD1B
-from .sdxl_turbo import SDXLTurboServe
+from .sdxl_turbo import ServeSDXLTurbo
 from .llama_cpp import ServeLlamaCpp

src/fastserve/models/__main__.py

@@ -1,7 +1,7 @@
 import argparse
 
 from fastserve.utils import get_default_device
-from . import SDXLTurboServe
+from . import ServeSDXLTurbo
 from .llama_cpp import ServeLlamaCpp
 from .ssd import ServeSSD1B
 
@@ -38,7 +38,7 @@
 if args.model == "ssd-1b":
     app = ServeSSD1B(device=device, timeout=args.timeout, batch_size=args.batch_size)
 elif args.model == "sdxl-turbo":
-    app = SDXLTurboServe(
+    app = ServeSDXLTurbo(
         device=device, timeout=args.timeout, batch_size=args.batch_size
     )
 

src/fastserve/models/sdxl_turbo.py

@@ -16,7 +16,7 @@ class PromptRequest(BaseModel):
     negative_prompt: str = "ugly, blurry, poor quality"
 
 
-class SDXLTurboServe(FastServe):
+class ServeSDXLTurbo(FastServe):
     def __init__(
         self, batch_size=1, timeout=0.0, device="cuda", num_inference_steps: int = 1
     ) -> None: