Staticaliza commited on
Commit
e334c60
·
verified ·
1 Parent(s): 0a0779d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -4
app.py CHANGED
@@ -11,7 +11,7 @@ import os
11
  import numpy as np
12
 
13
  from huggingface_hub import hf_hub_download
14
- from diffusers import DiffusionPipeline, StableDiffusionPipeline, DPMSolverMultistepScheduler
15
  from transformers import pipeline
16
  from PIL import Image
17
 
@@ -51,15 +51,12 @@ repo_default.load_lora_weights("nerijs/pixel-art-xl", adapter_name="pixel_base_2
51
  repo_pro = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16, use_safetensors=True)
52
  repo_pro.load_lora_weights(hf_hub_download("alimama-creative/FLUX.1-Turbo-Alpha", "diffusion_pytorch_model.safetensors"))
53
 
54
- # repo_classic = DiffusionPipeline.from_pretrained("gsdf/Counterfeit-V2.5", torch_dtype=torch.float16, use_safetensors=True, safety_checker=None)
55
-
56
  repo_customs = {
57
  "Default": repo_default,
58
  "Realistic": DiffusionPipeline.from_pretrained("ehristoforu/Visionix-alpha", torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False),
59
  "Anime": DiffusionPipeline.from_pretrained("cagliostrolab/animagine-xl-3.1", torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False),
60
  "Pixel": repo_default,
61
  "Pro": repo_pro,
62
- # "Classic": repo_classic,
63
  }
64
 
65
  # Functions
 
11
  import numpy as np
12
 
13
  from huggingface_hub import hf_hub_download
14
+ from diffusers import DiffusionPipeline
15
  from transformers import pipeline
16
  from PIL import Image
17
 
 
51
  repo_pro = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16, use_safetensors=True)
52
  repo_pro.load_lora_weights(hf_hub_download("alimama-creative/FLUX.1-Turbo-Alpha", "diffusion_pytorch_model.safetensors"))
53
 
 
 
54
  repo_customs = {
55
  "Default": repo_default,
56
  "Realistic": DiffusionPipeline.from_pretrained("ehristoforu/Visionix-alpha", torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False),
57
  "Anime": DiffusionPipeline.from_pretrained("cagliostrolab/animagine-xl-3.1", torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False),
58
  "Pixel": repo_default,
59
  "Pro": repo_pro,
 
60
  }
61
 
62
  # Functions