Update app.py
Browse files
app.py
CHANGED
|
@@ -36,7 +36,6 @@ 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", vae=vae, controlnet=controlnet, torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False)
|
| 42 |
|
|
@@ -110,8 +109,6 @@ def generate(input=DEFAULT_INPUT, filter_input="", negative_input=DEFAULT_NEGATI
|
|
| 110 |
"num_inference_steps": steps,
|
| 111 |
"guidance_scale": guidance,
|
| 112 |
"num_images_per_prompt": number,
|
| 113 |
-
"controlnet_conditioning_scale": 1,
|
| 114 |
-
"cross_attention_kwargs": {"scale": 1},
|
| 115 |
"generator": torch.Generator().manual_seed(seed),
|
| 116 |
"use_resolution_binning": True,
|
| 117 |
"output_type":"pil",
|
|
|
|
| 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 |
|
| 40 |
repo_default = StableDiffusionXLPipeline.from_pretrained("fluently/Fluently-XL-Final", vae=vae, controlnet=controlnet, torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False)
|
| 41 |
|
|
|
|
| 109 |
"num_inference_steps": steps,
|
| 110 |
"guidance_scale": guidance,
|
| 111 |
"num_images_per_prompt": number,
|
|
|
|
|
|
|
| 112 |
"generator": torch.Generator().manual_seed(seed),
|
| 113 |
"use_resolution_binning": True,
|
| 114 |
"output_type":"pil",
|