[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
opea-project · Sep 27, 2024 · d4ba8c7 · d4ba8c7
1 parent 045a11b
commit d4ba8c7
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 9 deletions.
diff --git a/comps/text2image/requirements.txt b/comps/text2image/requirements.txt
@@ -12,4 +12,4 @@ pydub
 shortuuid
 torch
 transformers
-uvicorn
+uvicorn
diff --git a/comps/text2image/text2image.py b/comps/text2image/text2image.py
@@ -21,7 +21,6 @@
     statistics_dict,
 )
 
-
 logger = CustomLogger("text2image")
 pipe = None
 args = None
@@ -42,12 +41,14 @@ def initialize():
             if not args.token:
                 args.token = os.getenv("HF_TOKEN")
             if args.device == "hpu":
-                kwargs.update({
-                    "use_habana": True,
-                    "use_hpu_graphs": args.use_hpu_graphs,
-                    "gaudi_config": "Habana/stable-diffusion",
-                    "token": args.token,
-                })
+                kwargs.update(
+                    {
+                        "use_habana": True,
+                        "use_hpu_graphs": args.use_hpu_graphs,
+                        "gaudi_config": "Habana/stable-diffusion",
+                        "token": args.token,
+                    }
+                )
                 # if "stable-diffusion-3" in args.model_name_or_path:
                 #     from optimum.habana.diffusers import GaudiStableDiffusion3Pipeline
 
@@ -117,6 +118,6 @@ def text2image(input: SDInputs):
     parser.add_argument("--bf16", action="store_true")
 
     args = parser.parse_args()
-    
+
     logger.info("Text2image server started.")
     opea_microservices["opea_service@text2image"].start()