Update app.py
Browse files
app.py
CHANGED
|
@@ -194,6 +194,7 @@ def load_txt2img(model_id):
|
|
| 194 |
print(f"⚠️ Fehler beim Laden der VAE: {vae_error}")
|
| 195 |
print("ℹ️ Versuche ohne VAE weiter...")
|
| 196 |
vae = None
|
|
|
|
| 197 |
|
| 198 |
# Modellparameter basierend auf Modelltyp
|
| 199 |
model_params = {
|
|
@@ -201,8 +202,10 @@ def load_txt2img(model_id):
|
|
| 201 |
"safety_checker": None,
|
| 202 |
"requires_safety_checker": False,
|
| 203 |
"add_watermarker": False,
|
| 204 |
-
"
|
| 205 |
-
}
|
|
|
|
|
|
|
| 206 |
|
| 207 |
# SAFETENSORS LOGIK HIER EINFÜGEN:
|
| 208 |
if model_id in SAFETENSORS_MODELS:
|
|
|
|
| 194 |
print(f"⚠️ Fehler beim Laden der VAE: {vae_error}")
|
| 195 |
print("ℹ️ Versuche ohne VAE weiter...")
|
| 196 |
vae = None
|
| 197 |
+
|
| 198 |
|
| 199 |
# Modellparameter basierend auf Modelltyp
|
| 200 |
model_params = {
|
|
|
|
| 202 |
"safety_checker": None,
|
| 203 |
"requires_safety_checker": False,
|
| 204 |
"add_watermarker": False,
|
| 205 |
+
"allow_pickle": True, # 1. Immer True für .bin
|
| 206 |
+
}
|
| 207 |
+
|
| 208 |
+
|
| 209 |
|
| 210 |
# SAFETENSORS LOGIK HIER EINFÜGEN:
|
| 211 |
if model_id in SAFETENSORS_MODELS:
|