Astridkraft commited on
Commit
9bc3391
·
verified ·
1 Parent(s): df058de

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
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
- "use_safetensors": True,
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: