Hadimeeee commited on
Commit
6e54e79
·
verified ·
1 Parent(s): c0fcf01

Upload pipeline.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. 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(lora_path)
 
 
 
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