Update app.py
Browse files
app.py
CHANGED
|
@@ -20,7 +20,7 @@ print(f"[SYSTEM] | Using {DEVICE} type compute device.")
|
|
| 20 |
# Variables
|
| 21 |
MAX_SEED = 9007199254740991
|
| 22 |
DEFAULT_INPUT = ""
|
| 23 |
-
DEFAULT_NEGATIVE_INPUT = "(deformed, distorted, disfigured:1.3),
|
| 24 |
DEFAULT_MODEL = "Default"
|
| 25 |
DEFAULT_HEIGHT = 1024
|
| 26 |
DEFAULT_WIDTH = 1024
|
|
@@ -36,13 +36,12 @@ footer {
|
|
| 36 |
repo_nsfw_classifier = pipeline("image-classification", model="Falconsai/nsfw_image_detection")
|
| 37 |
|
| 38 |
vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
|
| 39 |
-
controlnet = ControlNetModel.from_pretrained("MakiPan/controlnet-encoded-hands-130k", torch_dtype=torch.float16)
|
| 40 |
-
|
| 41 |
-
repo_default = StableDiffusionXLPipeline.from_pretrained("fluently/Fluently-XL-Final", torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False)
|
| 42 |
|
| 43 |
repo_default.load_lora_weights("ehristoforu/dalle-3-xl-v2", adapter_name="base")
|
| 44 |
repo_default.set_adapters("base")
|
| 45 |
-
# repo_default.set_adapters(["base"], adapter_weights=[0.7])
|
| 46 |
|
| 47 |
# repo_pixel = StableDiffusionXLPipeline.from_pretrained("sd-community/sdxl-flash", vae=vae, controlnet=controlnet, torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False)
|
| 48 |
# repo_pixel.load_lora_weights("artificialguybr/PixelArtRedmond", adapter_name="base")
|
|
|
|
| 20 |
# Variables
|
| 21 |
MAX_SEED = 9007199254740991
|
| 22 |
DEFAULT_INPUT = ""
|
| 23 |
+
DEFAULT_NEGATIVE_INPUT = "(bad), (abstract), (deformed, distorted, disfigured, disconnected, disgusting, displeasing:1.3), (anatomy, bad anatomy, wrong anatomy), (blur, blurry), (mutation, mutated), missing, unfinished, scribble, lowres, low quality, jpeg artifacts, text, error, username, scan, signature, watermark, (ugly), amputation, (digit, digits, limb, limbs, leg, legs, foot, feet, toe, toes, arm, arms, hand, hands, finger, fingers, head, heads:1.4), (exposed, explicit, porn, nude, nudity, naked, nsfw:1.25)"
|
| 24 |
DEFAULT_MODEL = "Default"
|
| 25 |
DEFAULT_HEIGHT = 1024
|
| 26 |
DEFAULT_WIDTH = 1024
|
|
|
|
| 36 |
repo_nsfw_classifier = pipeline("image-classification", model="Falconsai/nsfw_image_detection")
|
| 37 |
|
| 38 |
vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
|
| 39 |
+
controlnet = ControlNetModel.from_pretrained("MakiPan/controlnet-encoded-hands-130k", torch_dtype=torch.float16)
|
| 40 |
+
|
| 41 |
+
repo_default = StableDiffusionXLPipeline.from_pretrained("fluently/Fluently-XL-Final", vae=vae, controlnet=controlnet, torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False)
|
| 42 |
|
| 43 |
repo_default.load_lora_weights("ehristoforu/dalle-3-xl-v2", adapter_name="base")
|
| 44 |
repo_default.set_adapters("base")
|
|
|
|
| 45 |
|
| 46 |
# repo_pixel = StableDiffusionXLPipeline.from_pretrained("sd-community/sdxl-flash", vae=vae, controlnet=controlnet, torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False)
|
| 47 |
# repo_pixel.load_lora_weights("artificialguybr/PixelArtRedmond", adapter_name="base")
|