From 9b1112c5886d42b2d8ccfa8da9b061e5ad0308f2 Mon Sep 17 00:00:00 2001 From: Davidqian123 Date: Thu, 22 Aug 2024 05:47:31 +0000 Subject: [PATCH 1/2] fix bug --- nexa/cli/entry.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/nexa/cli/entry.py b/nexa/cli/entry.py index 66c3ca70..d8bdddb5 100644 --- a/nexa/cli/entry.py +++ b/nexa/cli/entry.py @@ -51,9 +51,9 @@ def run_ggml_inference(args): if hasattr(args, 'streamlit') and args.streamlit: inference.run_streamlit(model_path) elif args.img2img: - inference.loop_img2img() + inference.run_img2img() else: - inference.loop_txt2img() + inference.run_txt2img() return elif args.command == "vlm": from nexa.gguf.nexa_inference_vlm import NexaVLMInference From 711c1675fbe3c11ff8193e05f92e4585bc9c2979 Mon Sep 17 00:00:00 2001 From: Davidqian123 Date: Thu, 22 Aug 2024 05:48:24 +0000 Subject: [PATCH 2/2] update --- nexa/onnx/nexa_inference_image.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/nexa/onnx/nexa_inference_image.py b/nexa/onnx/nexa_inference_image.py index 87d1f959..4abb82fe 100644 --- a/nexa/onnx/nexa_inference_image.py +++ b/nexa/onnx/nexa_inference_image.py @@ -31,8 +31,8 @@ class NexaImageInference: A class used for loading image models and running image generation. Methods: - run: Run the image generation loop. - run_streamlit: Run the Streamlit UI. + run: Run the image generation loop. + run_streamlit: Run the Streamlit UI. Args: model_path (str): Path or identifier for the model in Nexa Model Hub.