Staticaliza commited on
Commit
23c51a5
·
verified ·
1 Parent(s): 43e6270

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -5
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), poorly drawn, bad anatomy, wrong anatomy, extra limb, missing limb, floating limbs, (mutated hands and fingers:1.4), disconnected limbs, mutation, mutated, ugly, disgusting, blurry, amputation, (NSFW:1.25)"
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
- # , vae=vae, controlnet=controlnet
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")