Update app.py

app.py

@@ -35,42 +35,9 @@ def genie (Model, Prompt, negative_prompt, height, width, scale, steps, seed):
         torch.cuda.empty_cache()
         return image
 
-    if Model == "SD3.5 Turbo":
-        torch.cuda.max_memory_allocated(device=device)
-        torch.cuda.empty_cache()
-        progress=gr.Progress(track_tqdm=True)
-        from diffusers import BitsAndBytesConfig, SD3Transformer2DModel
-        from diffusers import StableDiffusion3Pipeline
-
-
-        model_id = "stabilityai/stable-diffusion-3.5-large-turbo"
-
-        nf4_config = BitsAndBytesConfig(
-            load_in_4bit=True,
-            bnb_4bit_quant_type="nf4",
-            bnb_4bit_compute_dtype=torch.bfloat16)
-        model_nf4 = SD3Transformer2DModel.from_pretrained(
-            model_id,
-            subfolder="transformer",
-            quantization_config=nf4_config,
-            torch_dtype=torch.bfloat16)
-
-        t5_nf4 = T5EncoderModel.from_pretrained("diffusers/t5-nf4", torch_dtype=torch.bfloat16)
-
-        pipe = StableDiffusion3Pipeline.from_pretrained(
-            model_id, 
-            transformer=model_nf4,
-            text_encoder_3=t5_nf4,
-            torch_dtype=torch.bfloat16)
-        pipe.enable_model_cpu_offload()
-        pipe.vae.enable_slicing()
-        pipe.vae.enable_tiling()
-        torch.cuda.empty_cache()
-        image = pipe(prompt=Prompt, negative_prompt=negative_prompt, guidance_scale=0.0, num_inference_steps=4, width=1024, height=1024, generator=generator,).images[0]
-        torch.cuda.empty_cache()
     return image
     
-gr.Interface(fn=genie, inputs=[gr.Radio(['PhotoReal', 'Animagine XL 4', "SD3.5 Turbo"], value='PhotoReal', label='Choose Model'),
+gr.Interface(fn=genie, inputs=[gr.Radio(['PhotoReal', 'Animagine XL 4',], value='PhotoReal', label='Choose Model'),
                                gr.Textbox(label='What you want the AI to generate. 77 Token Limit.'), 
                                gr.Textbox(label='What you Do Not want the AI to generate. 77 Token Limit'),
                                gr.Slider(512, 1024, 768, step=128, label='Height'),