Update app.py

app.py

@@ -40,8 +40,9 @@ def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
 
 from diffusers import DiffusionPipeline
 
+# Use torch.float16 for better stability on NVIDIA GPUs
 base_model = "black-forest-labs/FLUX.1-dev"
-pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
+pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.float16)
 lora_repo = "strangerzonehf/Flux-Super-Realism-LoRA"
 trigger_word = "Super Realism"  # Leave blank if no trigger word is needed.
 pipe.load_lora_weights(lora_repo)
@@ -89,6 +90,8 @@ def generate_image_flux(
     positive_prompt = apply_style(style_name, prompt)
     if trigger_word:
         positive_prompt = f"{trigger_word} {positive_prompt}"
+    # Clear GPU cache before generating to help avoid memory fragmentation errors.
+    torch.cuda.empty_cache()
     images = pipe(
         prompt=positive_prompt,
         width=width,