Update app.py
Browse files
app.py
CHANGED
|
@@ -37,8 +37,8 @@ repo_nsfw_classifier = pipeline("image-classification", model="Falconsai/nsfw_im
|
|
| 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",
|
| 42 |
|
| 43 |
repo_default.load_lora_weights("ehristoforu/dalle-3-xl-v2", adapter_name="base")
|
| 44 |
repo_default.set_adapters("base")
|
|
@@ -56,14 +56,6 @@ repo_customs = {
|
|
| 56 |
"Pixel": None, # repo_pixel,
|
| 57 |
}
|
| 58 |
|
| 59 |
-
print("-----------A")
|
| 60 |
-
print(repo_default.scheduler)
|
| 61 |
-
print("-----/-////B")
|
| 62 |
-
print(EulerAncestralDiscreteScheduler.from_config(repo_default.scheduler.config))
|
| 63 |
-
print("-----------Z")
|
| 64 |
-
# repo_default.scheduler = EulerAncestralDiscreteScheduler.from_config(repo_default.scheduler.config)
|
| 65 |
-
|
| 66 |
-
|
| 67 |
# Functions
|
| 68 |
def save_image(img, seed):
|
| 69 |
name = f"{seed}-{uuid.uuid4()}.png"
|
|
|
|
| 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")
|
|
|
|
| 56 |
"Pixel": None, # repo_pixel,
|
| 57 |
}
|
| 58 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 59 |
# Functions
|
| 60 |
def save_image(img, seed):
|
| 61 |
name = f"{seed}-{uuid.uuid4()}.png"
|