Staticaliza commited on
Commit
5315b95
·
verified ·
1 Parent(s): b7c4387

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -46,14 +46,14 @@ repo_default.load_lora_weights("ehristoforu/dalle-3-xl-v2", adapter_name="defaul
46
  repo_default.load_lora_weights("artificialguybr/PixelArtRedmond", adapter_name="pixel_base")
47
  repo_default.load_lora_weights("nerijs/pixel-art-xl", adapter_name="pixel_base_2")
48
 
49
- # repo_pro = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16, use_safetensors=True, add_watermarker=False)
50
 
51
  repo_customs = {
52
  "Default": repo_default,
53
  "Realistic": DiffusionPipeline.from_pretrained("ehristoforu/Visionix-alpha", torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False),
54
  "Anime": DiffusionPipeline.from_pretrained("cagliostrolab/animagine-xl-3.1", torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False),
55
  "Pixel": repo_default,
56
- "Pro": None, # repo_pro,
57
  }
58
 
59
  # Functions
 
46
  repo_default.load_lora_weights("artificialguybr/PixelArtRedmond", adapter_name="pixel_base")
47
  repo_default.load_lora_weights("nerijs/pixel-art-xl", adapter_name="pixel_base_2")
48
 
49
+ repo_pro = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16, use_safetensors=True, add_watermarker=False)
50
 
51
  repo_customs = {
52
  "Default": repo_default,
53
  "Realistic": DiffusionPipeline.from_pretrained("ehristoforu/Visionix-alpha", torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False),
54
  "Anime": DiffusionPipeline.from_pretrained("cagliostrolab/animagine-xl-3.1", torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False),
55
  "Pixel": repo_default,
56
+ "Pro": repo_pro,
57
  }
58
 
59
  # Functions