Spaces:
Sleeping
Sleeping
adamelliotfields
commited on
Commit
•
af8f4a3
1
Parent(s):
7c0274b
No variant for single file checkpoints
Browse files- 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 |
-
|
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
|