Spaces:
Running on Zero
Running on Zero
Upload infer/common_infer.py with huggingface_hub
Browse files- infer/common_infer.py +1 -1
infer/common_infer.py
CHANGED
|
@@ -184,4 +184,4 @@ def initialize_pipeline(config):
|
|
| 184 |
print(f"[INFO] Loading trained LoRA from {lora_ckpt}...", flush=True)
|
| 185 |
pipeline.load_lora_weights(lora_ckpt, adapter_name="layer")
|
| 186 |
|
| 187 |
-
return pipeline, transp_vae
|
|
|
|
| 184 |
print(f"[INFO] Loading trained LoRA from {lora_ckpt}...", flush=True)
|
| 185 |
pipeline.load_lora_weights(lora_ckpt, adapter_name="layer")
|
| 186 |
|
| 187 |
+
return pipeline, transp_vae
|