Update app.py
Browse files
app.py
CHANGED
|
@@ -8,7 +8,6 @@ import os
|
|
| 8 |
from transformers import pipeline
|
| 9 |
|
| 10 |
from diffusers import StableDiffusionXLPipeline
|
| 11 |
-
from diffusers.models import AutoencoderKL
|
| 12 |
from PIL import Image
|
| 13 |
|
| 14 |
# Pre-Initialize
|
|
@@ -35,8 +34,6 @@ footer {
|
|
| 35 |
|
| 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", torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False)
|
| 41 |
repo_default.load_lora_weights("ehristoforu/dalle-3-xl-v2", adapter_name="base")
|
| 42 |
repo_default.set_adapters("base")
|
|
|
|
| 8 |
from transformers import pipeline
|
| 9 |
|
| 10 |
from diffusers import StableDiffusionXLPipeline
|
|
|
|
| 11 |
from PIL import Image
|
| 12 |
|
| 13 |
# Pre-Initialize
|
|
|
|
| 34 |
|
| 35 |
repo_nsfw_classifier = pipeline("image-classification", model="Falconsai/nsfw_image_detection")
|
| 36 |
|
|
|
|
|
|
|
| 37 |
repo_default = StableDiffusionXLPipeline.from_pretrained("fluently/Fluently-XL-Final", torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False)
|
| 38 |
repo_default.load_lora_weights("ehristoforu/dalle-3-xl-v2", adapter_name="base")
|
| 39 |
repo_default.set_adapters("base")
|