GiorgioV commited on
Commit
f1ead13
·
verified ·
1 Parent(s): d4dba0d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -51,7 +51,7 @@ def get_weights():
51
  weight_paths = get_weights()
52
 
53
  # --- MODEL INITIALIZATION ---
54
- model: ModelConfig = all_model_cfg['nsfw_gold_8.5k_final']
55
  model.model_path = weight_paths["model"]
56
  model.vae_path = weight_paths["vae"]
57
  model.synchformer_ckpt = weight_paths["sync"]
 
51
  weight_paths = get_weights()
52
 
53
  # --- MODEL INITIALIZATION ---
54
+ model: ModelConfig = all_model_cfg['large_44k']
55
  model.model_path = weight_paths["model"]
56
  model.vae_path = weight_paths["vae"]
57
  model.synchformer_ckpt = weight_paths["sync"]