Update app.py

app.py

@@ -47,30 +47,23 @@ repo_default = StableDiffusionXLPipeline.from_pretrained("fluently/Fluently-XL-F
 repo_default.load_lora_weights("ehristoforu/dalle-3-xl-v2", adapter_name="base")
 repo_default.set_adapters(["base"], adapter_weights=[0.7])
 
-repo_turbo = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16, use_safetensors=True, add_watermarker=False, revision="refs/pr/1").to(device)
-repo_turbo.load_lora_weights("ehristoforu/dalle-3-xl-v2", adapter_name="base")
-repo_turbo.set_adapters(["base"], adapter_weights=[0.7])
+repo_neo = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16, use_safetensors=True, add_watermarker=False, revision="refs/pr/1").to(device)
+repo_neo.load_lora_weights("ehristoforu/dalle-3-xl-v2", adapter_name="base")
+repo_neo.set_adapters(["base"], adapter_weights=[0.7])
 
 repo_pixel = StableDiffusionXLPipeline.from_pretrained("fluently/Fluently-XL-Final", torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False)
 repo_pixel.load_lora_weights("artificialguybr/PixelArtRedmond", adapter_name="base")
 repo_pixel.load_lora_weights("nerijs/pixel-art-xl", adapter_name="base2")
 repo_pixel.set_adapters(["base", "base2"], adapter_weights=[1, 1])
 
-# repo_large_path = snapshot_download(repo_id="stabilityai/stable-diffusion-3-medium", revision="refs/pr/26", token=HF_TOKEN)
-# repo_large_transformer_path = SD3Transformer2DModel.from_pretrained(repo_large_path, subfolder="transformer", torch_dtype=torch.float16)
-# repo_large_transformer = PeftModel.from_pretrained(repo_large_transformer_path, "jasperai/flash-sd3")
-
 repo_customs = {
     "Default": repo_default,
-    "Turbo": repo_turbo,
+    "Neo": repo_neo,
     "Realistic": StableDiffusionXLPipeline.from_pretrained("ehristoforu/Visionix-alpha", torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False),
     "Anime": StableDiffusionXLPipeline.from_pretrained("cagliostrolab/animagine-xl-3.1", torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False),
     "Pixel": repo_pixel,
-    "Large": None, # StableDiffusion3Pipeline.from_pretrained(repo_large_path, transformer=repo_large_transformer, torch_dtype=torch.float16, use_safetensors=True),
 }
 
-# repo_customs["Large"].scheduler = FlashFlowMatchEulerDiscreteScheduler.from_pretrained(repo_large_path, subfolder="scheduler")
-
 # Functions
 def save_image(img, seed):
     name = f"{seed}-{uuid.uuid4()}.png"
@@ -99,7 +92,7 @@ def generate(input=DEFAULT_INPUT, filter_input="", negative_input=DEFAULT_NEGATI
     if model == "Realistic":   
         steps_set = 25
         guidance_set = 7
-    if model == "Turbo":
+    if model == "Neo":
         steps_set = 4
         guidance_set = 0
     elif model == "Anime":   
@@ -108,9 +101,6 @@ def generate(input=DEFAULT_INPUT, filter_input="", negative_input=DEFAULT_NEGATI
     elif model == "Pixel":   
         steps_set = 15
         guidance_set = 1.5
-    elif model == "Large":   
-        steps_set = 15
-        guidance_set = 1.5
     else:
         steps_set = 25
         guidance_set = 7