https://github.com/byrkbrk/generating-by-prompt-flux-schnell

Generate photo-realistic & high-resolution images by user-defined prompts using Flux-schnell, in PyTorch & Gradio
https://github.com/byrkbrk/generating-by-prompt-flux-schnell

flux-schnell generative-art huggingface-diffusers pytorch text-to-image

Last synced: about 5 hours ago
JSON representation

Generate photo-realistic & high-resolution images by user-defined prompts using Flux-schnell, in PyTorch & Gradio

• Host: GitHub
• URL: https://github.com/byrkbrk/generating-by-prompt-flux-schnell
• Owner: byrkbrk
• Created: 2024-08-17T18:08:13.000Z (about 1 month ago)
• Default Branch: main
• Last Pushed: 2024-08-30T21:02:18.000Z (27 days ago)
• Last Synced: 2024-09-22T06:02:17.094Z (5 days ago)
• Topics: flux-schnell, generative-art, huggingface-diffusers, pytorch, text-to-image
• Language: Python
• Homepage:
• Size: 9.16 MB
• Stars: 0
• Watchers: 1
• Forks: 1
• Open Issues: 0
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

        
# Generate by Prompt using Flux-schnell

## Introduction

We design a module that generates photo-realistic & high resolution images based on user-defined prompts. While preparing the module,

we utilize the pretrained model [Flux-schnell  at Hugging Face](https://huggingface.co/black-forest-labs/FLUX.1-schnell) provided by [black forests labs](https://blackforestlabs.ai/).

## Setting Up the Environment

### Using Conda (recommended)

1. Install [Conda](https://conda.io/projects/conda/en/latest/user-guide/install/index.html), if not already installed.

2. Clone the repository:

    ~~~

    git clone https://github.com/byrkbrk/generating-by-prompt-flux-schnell.git

    ~~~

3. Change the directory:

    ~~~

    cd generating-by-prompt-flux-schnell

    ~~~

4. Create the environment:

    ~~~

    conda env create -f environment.yaml

    ~~~

5. Activate the environment:

    ~~~

    conda activate generating-by-prompt-flux-schnell

    ~~~

### Using pip

1. Download & install [Python](https://www.python.org/downloads/) (version==3.11)

2. Clone the repository:

    ~~~

    git clone https://github.com/byrkbrk/generating-by-prompt-flux-schnell.git

    ~~~

3. Change the directory:

    ~~~

    cd generating-by-prompt-flux-schnell

    ~~~

4. Install packages using `pip`:

    ~~~

    pip install -r requirements.txt

    ~~~

## Generating Images

Check it out how to use:

~~~

python3 generate.py --help

~~~

Output:

~~~

Generate images by prompt using Flux-schnell

positional arguments:

  prompt                Prompt that be used during inference

options:

  -h, --help            show this help message and exit

  --num_inference_steps NUM_INFERENCE_STEPS

                        Number of inference steps used during generating

  --device {cuda,mps,cpu}

                        The device used during inference. Default: `None`

  --enable_sequential_cpu_offload

                        Enables sequential cpu offload during inference

~~~

### Example usages

Execute the following code blocks to generate the corresponding images displayed below. The results will be saved into the folder `./generated-images`.

~~~

python3 generate.py\

 "an image of a turtle in Picasso style"\

 --num_inference_steps 4\

 --enable_sequential_cpu_offload

~~~



  



~~~

python3 generate.py\

 "an image of a turtle in Camille Pissarro style"\

 --num_inference_steps 4\

 --enable_sequential_cpu_offload

~~~



  



~~~

python3 generate.py\

 "an image of a turtle in Claude Monet style"\

 --num_inference_steps 4\

 --enable_sequential_cpu_offload

~~~



  



## Generating via Gradio

Check it out how to use:

~~~

python3 app.py --help

~~~

Output:

~~~

Generate image using Flux-schnell via Gradio

options:

  -h, --help            show this help message and exit

  --enable_sequential_cpu_offload

                        Enables sequential cpu offload

  --share               Allows Gradio to produce public link

~~~

### Example usage

To run the app on your local device, execute the following:

~~~

python3 app.py\

 --enable_sequential_cpu_offload

~~~

Then, visit the url [http://127.0.0.1:7860](http://127.0.0.1:7860) to open the interface displayed below: