adamelliotfields commited on
Commit
af8f4a3
1 Parent(s): 7c0274b

No variant for single file checkpoints

Browse files
Files changed (1) hide show
  1. lib/loader.py +1 -2
lib/loader.py CHANGED
@@ -156,8 +156,7 @@ class Loader:
156
  scheduler_kwargs["clip_sample"] = False
157
  scheduler_kwargs["set_alpha_to_one"] = False
158
 
159
- # no fp16 variant (already half-precision)
160
- if model_lower not in ["cagliostrolab/animagine-xl-3.1", "fluently/fluently-xl-final"]:
161
  variant = "fp16"
162
  else:
163
  variant = None
 
156
  scheduler_kwargs["clip_sample"] = False
157
  scheduler_kwargs["set_alpha_to_one"] = False
158
 
159
+ if model_lower not in Config.MODEL_CHECKPOINTS.keys():
 
160
  variant = "fp16"
161
  else:
162
  variant = None