more debug
Browse files
models/pipeline_loader.py
CHANGED
|
@@ -19,7 +19,10 @@ def get_pipeline(model_path):
|
|
| 19 |
# Assume it's a Hugging Face Hub model ID
|
| 20 |
# Try to load config to determine if it's text-conditional
|
| 21 |
print(model_path)
|
| 22 |
-
|
|
|
|
|
|
|
|
|
|
| 23 |
has_text_encoder = "text_encoder" in config
|
| 24 |
|
| 25 |
if has_text_encoder:
|
|
|
|
| 19 |
# Assume it's a Hugging Face Hub model ID
|
| 20 |
# Try to load config to determine if it's text-conditional
|
| 21 |
print(model_path)
|
| 22 |
+
print("\n\n\n\n")
|
| 23 |
+
config = DiffusionPipeline.load_config(model_path)
|
| 24 |
+
print(config)
|
| 25 |
+
print("\n\n\n\n")
|
| 26 |
has_text_encoder = "text_encoder" in config
|
| 27 |
|
| 28 |
if has_text_encoder:
|