Spaces:
Running
on
L40S
Running
on
L40S
Rishi Desai
commited on
Commit
·
453e82b
1
Parent(s):
da41944
more readme edits
Browse files- README.md +3 -13
- face_enhance.py +0 -4
README.md
CHANGED
@@ -25,14 +25,8 @@ A tool for improving facial consistency and quality in AI-generated images. Dram
|
|
25 |
|
26 |
### Setup
|
27 |
|
28 |
-
1.
|
29 |
-
<details>
|
30 |
-
<summary>Instructions</summary>
|
31 |
-
|
32 |
-
- Log into Hugging Face and accept their terms of service to download [Flux.1-dev](https://huggingface.co/black-forest-labs/FLUX.1-dev)
|
33 |
-
- Models will be downloaded to `HF_HOME` and symlinked to `FaceEnhance/ComfyUI/models/`
|
34 |
|
35 |
-
</details>
|
36 |
|
37 |
2. Create the virtual environment:
|
38 |
```bash
|
@@ -58,7 +52,7 @@ A tool for improving facial consistency and quality in AI-generated images. Dram
|
|
58 |
|
59 |
This will
|
60 |
- Install ComfyUI, custom nodes, and remaining dependencies to your venv
|
61 |
-
- Download all
|
62 |
|
63 |
4. Run inference on one example:
|
64 |
|
@@ -77,11 +71,7 @@ A tool for improving facial consistency and quality in AI-generated images. Dram
|
|
77 |
|
78 |
## Gradio Demo
|
79 |
|
80 |
-
A simple web interface for the face enhancement workflow.
|
81 |
-
|
82 |
-
1. Run `python demo.py`
|
83 |
-
|
84 |
-
2. Go to http://localhost:7860. You may need to enable port forwarding.
|
85 |
|
86 |
## ComfyUI
|
87 |
|
|
|
25 |
|
26 |
### Setup
|
27 |
|
28 |
+
1. Log into Hugging Face and accept their terms of service to download [Flux.1-dev](https://huggingface.co/black-forest-labs/FLUX.1-dev)
|
|
|
|
|
|
|
|
|
|
|
29 |
|
|
|
30 |
|
31 |
2. Create the virtual environment:
|
32 |
```bash
|
|
|
52 |
|
53 |
This will
|
54 |
- Install ComfyUI, custom nodes, and remaining dependencies to your venv
|
55 |
+
- Download all models to `HF_HOME` and create symlinks to `FaceEnhance/ComfyUI/models/`
|
56 |
|
57 |
4. Run inference on one example:
|
58 |
|
|
|
71 |
|
72 |
## Gradio Demo
|
73 |
|
74 |
+
A simple web interface for the face enhancement workflow. Run `python demo.py`
|
|
|
|
|
|
|
|
|
75 |
|
76 |
## ComfyUI
|
77 |
|
face_enhance.py
CHANGED
@@ -236,10 +236,6 @@ def main(
|
|
236 |
basicscheduler = NODE_CLASS_MAPPINGS["BasicScheduler"]()
|
237 |
samplercustomadvanced = NODE_CLASS_MAPPINGS["SamplerCustomAdvanced"]()
|
238 |
vaedecode = VAEDecode()
|
239 |
-
# faceembeddistance = NODE_CLASS_MAPPINGS["FaceEmbedDistance"]()
|
240 |
-
# display_any_rgthree = NODE_CLASS_MAPPINGS["Display Any (rgthree)"]()
|
241 |
-
# image_comparer_rgthree = NODE_CLASS_MAPPINGS["Image Comparer (rgthree)"]()
|
242 |
-
# saveimage = SaveImage()
|
243 |
|
244 |
applypulidflux_133 = applypulidflux.apply_pulid_flux(
|
245 |
weight=id_weight,
|
|
|
236 |
basicscheduler = NODE_CLASS_MAPPINGS["BasicScheduler"]()
|
237 |
samplercustomadvanced = NODE_CLASS_MAPPINGS["SamplerCustomAdvanced"]()
|
238 |
vaedecode = VAEDecode()
|
|
|
|
|
|
|
|
|
239 |
|
240 |
applypulidflux_133 = applypulidflux.apply_pulid_flux(
|
241 |
weight=id_weight,
|