Update inference_manager.py
Browse files- inference_manager.py +1 -1
inference_manager.py
CHANGED
@@ -216,7 +216,7 @@ class InferenceManager:
|
|
216 |
#pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
|
217 |
clip_skip = cfg.get("clip_skip", 1)
|
218 |
# Adjust clip skip for XL (assumed not relevant for SD 1.5)
|
219 |
-
pipe.text_encoder.config.num_hidden_layers -= (clip_skip - 1)
|
220 |
|
221 |
load_time = round(time.time() - start, 2)
|
222 |
print(f"Base model loaded in {load_time}s")
|
|
|
216 |
#pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
|
217 |
clip_skip = cfg.get("clip_skip", 1)
|
218 |
# Adjust clip skip for XL (assumed not relevant for SD 1.5)
|
219 |
+
#pipe.text_encoder.config.num_hidden_layers -= (clip_skip - 1)#!TEST! remove skip clip
|
220 |
|
221 |
load_time = round(time.time() - start, 2)
|
222 |
print(f"Base model loaded in {load_time}s")
|