Update app.py

app.py

@@ -41,7 +41,7 @@ def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
 from diffusers import DiffusionPipeline
 
 base_model = "black-forest-labs/FLUX.1-dev"
-pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype="auto")
+pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
 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)
@@ -118,7 +118,7 @@ smol_processor = AutoProcessor.from_pretrained("HuggingFaceTB/SmolVLM2-2.2B-Inst
 smol_model = AutoModelForImageTextToText.from_pretrained(
     "HuggingFaceTB/SmolVLM2-2.2B-Instruct", 
     _attn_implementation="flash_attention_2",
-    torch_dtype="auto"
+    torch_dtype=torch.bfloat16
 ).to("cuda:0")
 
 # -------------------------------