Update app.py

app.py

@@ -62,6 +62,13 @@ MODEL_CONFIGS = {
     }
 }
 
+
+# === SAFETENSORS KONFIGURATION ===
+SAFETENSORS_MODELS = [
+    "runwayml/stable-diffusion-v1-5",
+    "stabilityai/stable-diffusion-2-1"
+]
+
 # Aktuell ausgewähltes Modell (wird vom User gesetzt)
 current_model_id = "runwayml/stable-diffusion-v1-5"
 
@@ -196,6 +203,15 @@ def load_txt2img(model_id):
             "add_watermarker": False,
             "use_safetensors": True,
         }
+
+        # SAFETENSORS LOGIK HIER EINFÜGEN:
+        if model_id in SAFETENSORS_MODELS:
+            model_params["use_safetensors"] = True
+            print(f"ℹ️ Verwende safetensors für {model_id}")
+        else:
+            model_params["use_safetensors"] = False
+            print(f"ℹ️ Verwende .bin weights für {model_id}")
+     
         
         # NUR FP16 Variante laden wenn Modell sie unterstützt UND wir auf GPU sind
         if config.get("supports_fp16", False) and torch_dtype == torch.float16: