init

app.py

@@ -0,0 +1,92 @@
+from diffusers import DiffusionPipeline, ControlNetModel, StableDiffusionXLControlNetPipeline, AutoencoderKL
+from diffusers.utils import load_image
+from PIL import Image
+import os, time
+import torch
+import numpy as np
+import cv2
+import gradio as gr
+from torchvision import transforms 
+
+lora_path = "/home/user/lora"
+output_dir = "/home/user/output"
+
+model = DiffusionPipeline.from_pretrained(
+    "SimianLuo/LCM_Dreamshaper_v7",
+    safety_checker = None,
+)
+model.to(torch_device="cpu", torch_dtype=torch.float32).to("mps")
+
+low_threshold = 100
+high_threshold = 200
+
+safetensors_files_without_suffix = []
+
+def process(prompt, negative_prompt, width, height, steps, scale, number, loras):
+    if len(loras) > 0:
+        for item in loras:
+            model.load_lora_weights(lora_path, weight_name=f"{item}.safetensors")
+        
+    for _ in range(number):
+        _seed = int.from_bytes(os.urandom(2), "big")
+        generator = torch.manual_seed(_seed)
+        image = model(
+            prompt=prompt, 
+            negative_prompt=negative_prompt,
+            width=width, 
+            height=height,
+            num_inference_steps=steps,
+            generator=generator,
+            guidance_scale=scale
+        ).images[0]
+
+        timestamp = time.strftime("%Y%m%d-%H%M%S")
+        output_path = os.path.join(output_dir, f"{timestamp}.png")
+        image.save(output_path)
+
+        # images.append(image)
+    # return images
+
+block = gr.Blocks().queue()
+
+def update_loras_list():
+    filenames = os.listdir(lora_path)
+    safetensors_files = [filename for filename in filenames if filename.endswith(".safetensors")]
+    safetensors_files_without_suffix = [filename[:-len(".safetensors")] for filename in safetensors_files]
+    return gr.Dropdown.update(choices=safetensors_files_without_suffix)
+
+with gr.Blocks(css="style.css") as block:
+    with gr.Row():
+        with gr.Column():
+            prompt = gr.Textbox(label="正向提示词", lines=3, value="Warhammer 40k, a new robot in Armor, god of empire detailed face octane painting, Devian art, concept art, The gate of the medieval castle")
+            negative_prompt = gr.Textbox(label="反向提示词", value="low quality")
+            loras = gr.Dropdown(safetensors_files_without_suffix, multiselect=True, label="Loras", info="增加Loras风格")
+            with gr.Row():
+                load_loras_btn = gr.Button(value="加载")
+                clear_btn = gr.ClearButton(components=[loras], value="清空")
+                    
+        with gr.Column():
+            with gr.Row():
+                steps = gr.Slider(label="迭代步数", minimum=1, maximum=50, value=4, step=1)
+                scale = gr.Slider(label="引导系数", minimum=1, maximum=50, value=4, step=0.5)
+                number = gr.Slider(label="生成数量", minimum=1, maximum=10, value=1, step=1)
+
+            with gr.Row():
+                width = gr.Slider(label="宽度", minimum=384, maximum=1024, value=512, step=128)
+                height = gr.Slider(label="高度", minimum=384, maximum=1024, value=512, step=128)
+
+            with gr.Row():
+                # result_gallery = gr.Gallery(label='输出', show_label=True, elem_id="gallery", columns=[2], height='auto')
+                run_button = gr.Button(value="运行", elem_id="blue-button",)
+
+    ips = [prompt, negative_prompt, width, height, steps, scale, number, loras]
+    run_button.click(fn=process, inputs=ips)
+    load_loras_btn.click(fn=update_loras_list, outputs=loras)
+
+
+def main():
+    block.launch(debug=True, share=False)
+
+if __name__ == "__main__":
+    main()
+    

requirements.txt

@@ -0,0 +1,229 @@
+absl-py==2.0.0
+accelerate==0.23.0
+addict==2.4.0
+aenum==3.1.15
+aiofiles==23.2.1
+aiohttp==3.9.1
+aiosignal==1.3.1
+albumentations==1.3.1
+altair==5.2.0
+altgraph==0.17.3
+antlr4-python3-runtime==4.9.3
+anyio==3.7.1
+argostranslate==1.9.1
+attrs==23.1.0
+basicsr==1.4.2
+beautifulsoup4==4.12.2
+blendmodes==2022
+blessed==1.20.0
+blinker==1.6.2
+cachetools==5.3.2
+certifi==2023.11.17
+cffi==1.16.0
+chardet==5.2.0
+charset-normalizer==3.3.2
+clean-fid==0.1.35
+click==8.1.5
+clip @ https://github.com/openai/CLIP/archive/d50d76daa670286dd6cacf3bcd80b5e4823fc8e1.zip#sha256=b5842c25da441d6c581b53a5c60e0c2127ebafe0f746f8e15561a006c6c3be6a
+clip-interrogator==0.6.0
+cmake==3.28.1
+coloredlogs==15.0.1
+colorlog==6.8.0
+contourpy==1.2.0
+cssselect2==0.7.0
+cstr @ git+https://github.com/WASasquatch/cstr@0520c29a18a7a869a6e5983861d6f7a4c86f8e9b
+ctranslate2==3.20.0
+cycler==0.12.1
+Cython==3.0.7
+deep-translator==1.11.4
+deprecation==2.1.0
+diffusers==0.23.0
+easydict==1.11
+einops==0.4.1
+enlighten==1.12.2
+facexlib==0.3.0
+fairscale==0.4.13
+fastapi==0.94.0
+ffmpy @ git+https://github.com/WASasquatch/ffmpy.git@f000737698b387ffaeab7cd871b0e9185811230d
+filelock==3.13.1
+filterpy==1.4.5
+Flask==2.3.2
+Flask-JWT-Extended==4.5.2
+Flask-SQLAlchemy==3.0.5
+Flask-WTF==1.1.1
+flatbuffers==23.5.26
+fonttools==4.47.0
+frozenlist==1.4.1
+fsspec==2023.12.2
+ftfy==6.1.3
+future==0.18.3
+fvcore==0.1.5.post20221221
+gdown==4.7.1
+gevent==23.7.0
+gfpgan==1.3.8
+gitdb==4.0.11
+GitPython==3.1.32
+google-auth==2.25.2
+google-auth-oauthlib==1.2.0
+googletrans-py==4.0.0
+gradio==3.41.2
+gradio_client==0.5.0
+greenlet==2.0.2
+grpcio==1.60.0
+h11==0.12.0
+h2==4.1.0
+hpack==4.0.0
+httpcore==0.15.0
+httpx==0.24.1
+huggingface-hub==0.17.3
+humanfriendly==10.0
+hyperframe==6.0.1
+idna==3.6
+imageio==2.33.1
+imageio-ffmpeg==0.4.9
+img2texture @ git+https://github.com/WASasquatch/img2texture.git@a546609a915caf1dcb84a15bf44b5b639ad924e6
+importlib-metadata==7.0.1
+importlib-resources==6.1.1
+inflection==0.5.1
+insightface==0.7.3
+iopath==0.1.9
+itsdangerous==2.1.2
+Jinja2==3.1.2
+joblib==1.3.2
+jsonmerge==1.8.0
+jsonschema==4.20.0
+jsonschema-specifications==2023.12.1
+kiwisolver==1.4.5
+kornia==0.6.7
+lark==1.1.2
+lazy_loader==0.3
+lightning-utilities==0.10.0
+llvmlite==0.41.1
+lmdb==1.4.1
+lpips==0.1.4
+lxml==4.9.4
+macholib==1.16.2
+mapbox-earcut==1.0.1
+Markdown==3.5.1
+MarkupSafe==2.1.3
+matplotlib==3.8.2
+mediapipe==0.10.9
+mpmath==1.3.0
+multidict==6.0.4
+networkx==3.2.1
+nsz==4.5.0
+numba==0.58.1
+numpy==1.24.4
+oauthlib==3.2.2
+omegaconf==2.2.3
+onnx==1.15.0
+onnxruntime==1.16.3
+open-clip-torch==2.20.0
+opencv-contrib-python==4.8.1.78
+opencv-python==4.8.1.78
+opencv-python-headless==4.7.0.72
+orjson==3.9.10
+packaging==23.2
+pandas==2.1.4
+piexif==1.1.3
+pilgram==1.2.1
+Pillow==10.1.0
+platformdirs==4.1.0
+pooch==1.8.0
+portalocker==2.8.2
+prefixed==0.7.0
+prettytable==3.9.0
+protobuf==3.20.3
+psutil==5.9.5
+pyasn1==0.5.1
+pyasn1-modules==0.3.0
+pycollada==0.7.2
+pycparser==2.21
+pycryptodome==3.19.0
+pydantic==1.10.13
+pydub==0.25.1
+pyinstaller==5.13.0
+pyinstaller-hooks-contrib==2023.5
+PyJWT==2.7.0
+PyMatting==1.1.12
+PyMySQL==1.1.0
+pyparsing==3.1.1
+PySocks==1.7.1
+python-dateutil==2.8.2
+python-multipart==0.0.6
+pytorch-lightning==1.9.4
+pytz==2023.3.post1
+pywatchman==1.4.1
+PyWavelets==1.5.0
+PyYAML==6.0.1
+qudida==0.0.4
+realesrgan==0.3.0
+referencing==0.32.0
+regex==2023.12.25
+rembg==2.0.53
+reportlab==4.0.8
+requests==2.31.0
+requests-oauthlib==1.3.1
+resize-right==0.0.2
+rpds-py==0.16.2
+rsa==4.9
+Rtree==1.1.0
+sacremoses==0.0.53
+safetensors==0.3.1
+scikit-image==0.20.0
+scikit-learn==1.3.2
+scipy==1.11.4
+semantic-version==2.10.0
+sentencepiece==0.1.99
+shapely==2.0.2
+simpleeval==0.9.13
+six==1.16.0
+smmap==5.0.1
+sniffio==1.3.0
+sounddevice==0.4.6
+soupsieve==2.5
+SQLAlchemy==2.0.18
+stanza==1.1.1
+starlette==0.26.1
+svg.path==6.3
+svglib==1.5.1
+sympy==1.12
+tabulate==0.9.0
+tb-nightly==2.16.0a20231229
+tensorboard-data-server==0.7.2
+termcolor==2.4.0
+tf_keras-nightly==2.16.0.dev2023122910
+threadpoolctl==3.2.0
+tifffile==2023.12.9
+timm==0.9.2
+tinycss2==1.2.1
+tokenizers==0.14.1
+tomesd==0.1.3
+tomli==2.0.1
+toolz==0.12.0
+torch==2.0.1
+torchdiffeq==0.2.3
+torchmetrics==1.2.1
+torchsde==0.2.6
+torchvision==0.15.2
+tqdm==4.66.1
+trampoline==0.1.2
+transformers==4.34.1
+trimesh==3.21.7
+typing_extensions==4.7.1
+tzdata==2023.4
+urllib3==2.1.0
+uvicorn==0.25.0
+wcwidth==0.2.12
+webencodings==0.5.1
+websockets==11.0.3
+Werkzeug==2.3.6
+WTForms==3.0.1
+xxhash==3.4.1
+yacs==0.1.8
+yapf==0.40.2
+yarl==1.9.4
+zipp==3.17.0
+zope.event==5.0
+zope.interface==6.0
+zstandard==0.22.0