diff --git a/docs/developer/Http_Service.md b/docs/developer/Http_Service.md index e35e7e5..1f0b351 100644 --- a/docs/developer/Http_Service.md +++ b/docs/developer/Http_Service.md @@ -1,9 +1,11 @@ ## Launch a Text-to-Image Http Service -Launch an HTTP-based text-to-image service that generates images from textual descriptions (prompts) using the DiT model. The generated images can either be returned directly to users or saved to a specified disk location. +Launch an HTTP-based text-to-image service that generates images from textual descriptions (prompts) using the DiT model. +The generated images can either be returned directly to users or saved to a specified disk location. +For example, the following command launches a HTTP service with 4 GPUs, 2 Ulysses parallel degree, 2 PipeFusion parallel degree, and the model path is `./models/FLUX.1-schnell`. ```bash -python ./entrypoints/launch.py --world_size 4 --ulysses_parallel_degree 2 --pipefusion_parallel_degree 2 --model_path /your_model_path +python ./entrypoints/launch.py --world_size 4 --ulysses_parallel_degree 2 --pipefusion_parallel_degree 2 --model_path /your_model_path/FLUX.1-schnell ``` diff --git a/entrypoints/launch.py b/entrypoints/launch.py index 1cc7b7b..2f39f85 100644 --- a/entrypoints/launch.py +++ b/entrypoints/launch.py @@ -183,7 +183,7 @@ async def generate_image(request: GenerateRequest): if __name__ == "__main__": parser = argparse.ArgumentParser(description='xDiT HTTP Service') - parser.add_argument('--model_path', type=str, default="/demo-huabei2/models/dit/FLUX.1-schnell", help='Path to the model') + parser.add_argument('--model_path', type=str, help='Path to the model', required=True) parser.add_argument('--world_size', type=int, default=1, help='Number of parallel workers') parser.add_argument('--pipefusion_parallel_degree', type=int, default=1, help='Degree of pipeline fusion parallelism') parser.add_argument('--ulysses_parallel_degree', type=int, default=1, help='Degree of Ulysses parallelism') diff --git a/xfuser/__version__.py b/xfuser/__version__.py index 61eca0a..df12433 100644 --- a/xfuser/__version__.py +++ b/xfuser/__version__.py @@ -1 +1 @@ -__version__ = "0.4.2rc2" +__version__ = "0.4.2"