brightpay commited on
Commit
641d496
β€’
1 Parent(s): 1c4b07d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ from diffusers import StableDiffusionPipeline
3
  from safetensors.torch import load_file
4
  import torch
5
 
6
- def load_pipeline(base_model_path, lora_path):
7
  # Load base SD-v1.5 pipeline
8
  pipeline = StableDiffusionPipeline.from_pretrained(base_model_path)
9
 
 
3
  from safetensors.torch import load_file
4
  import torch
5
 
6
+ def load_pipeline(base_model_path, lora_repo_id, lora_filename):
7
  # Load base SD-v1.5 pipeline
8
  pipeline = StableDiffusionPipeline.from_pretrained(base_model_path)
9