Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Mar 19, 2024
1 parent f1088ee commit 817e90b
Show file tree
Hide file tree
Showing 3 changed files with 5 additions and 6 deletions.
6 changes: 3 additions & 3 deletions src/fastserve/models/__init__.py
Original file line number Diff line number Diff line change
@@ -1,9 +1,9 @@
from fastserve.models.cv.face_reco import FaceDetection as FaceDetection
from fastserve.models.llm.huggingface import ServeHuggingFace as ServeHuggingFace
from fastserve.models.cv.image_classification import (
ServeImageClassification as ServeImageClassification,
)
from fastserve.models.llm.llama_cpp import ServeLlamaCpp as ServeLlamaCpp
from fastserve.models.image_gen.sdxl_turbo import ServeSDXLTurbo as ServeSDXLTurbo
from fastserve.models.ssd import ServeSSD1B as ServeSSD1B
from fastserve.models.llm.huggingface import ServeHuggingFace as ServeHuggingFace
from fastserve.models.llm.llama_cpp import ServeLlamaCpp as ServeLlamaCpp
from fastserve.models.llm.vllm import ServeVLLM as ServeVLLM
from fastserve.models.ssd import ServeSSD1B as ServeSSD1B
2 changes: 1 addition & 1 deletion src/fastserve/models/__main__.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,9 +2,9 @@

from fastserve.models import ServeImageClassification
from fastserve.models.cv.face_reco import FaceDetection
from fastserve.models.image_gen.sdxl_turbo import ServeSDXLTurbo
from fastserve.models.llm.huggingface import ServeHuggingFace
from fastserve.models.llm.llama_cpp import ServeLlamaCpp
from fastserve.models.image_gen.sdxl_turbo import ServeSDXLTurbo
from fastserve.models.ssd import ServeSSD1B
from fastserve.utils import get_default_device

Expand Down
3 changes: 1 addition & 2 deletions src/fastserve/models/image_gen/sdxl_turbo.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,10 +6,9 @@
import torch
from diffusers import AutoPipelineForText2Image
from fastapi.responses import StreamingResponse
from pydantic import BaseModel

from fastserve import FastServe
from fastserve.utils import get_ui_folder
from pydantic import BaseModel


class PromptRequest(BaseModel):
Expand Down

0 comments on commit 817e90b

Please sign in to comment.