Staticaliza commited on
Commit
80bd51f
·
verified ·
1 Parent(s): 64545fb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -51,12 +51,15 @@ 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_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
 
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