Upload pipeline.py with huggingface_hub
Browse files- pipeline.py +4 -1
pipeline.py
CHANGED
|
@@ -57,7 +57,10 @@ class Mongle32BitPipeline:
|
|
| 57 |
torch_dtype=self.dtype,
|
| 58 |
use_safetensors=True,
|
| 59 |
)
|
| 60 |
-
self.pipe.load_lora_weights(
|
|
|
|
|
|
|
|
|
|
| 61 |
self.pipe.to(self.device)
|
| 62 |
self.pipe.enable_attention_slicing()
|
| 63 |
|
|
|
|
| 57 |
torch_dtype=self.dtype,
|
| 58 |
use_safetensors=True,
|
| 59 |
)
|
| 60 |
+
self.pipe.load_lora_weights(
|
| 61 |
+
lora_path,
|
| 62 |
+
weight_name="pytorch_lora_weights.safetensors",
|
| 63 |
+
)
|
| 64 |
self.pipe.to(self.device)
|
| 65 |
self.pipe.enable_attention_slicing()
|
| 66 |
|