PseudoTerminal X commited on
Commit
a5590b5
1 Parent(s): 17b9398

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -155,7 +155,7 @@ def generate_image(pipeline, prompt, output_file, num_inference_steps, width, he
155
  def main():
156
  parser = argparse.ArgumentParser(description="Generate images using a custom diffusion pipeline with LoRA weights.")
157
  parser.add_argument("--model_id", type=str, default='black-forest-labs/FLUX.1-dev', help="Model ID from Hugging Face Hub.")
158
- parser.add_argument("--adapter_id", type=str, required=True, help="LoRA weights file.")
159
  parser.add_argument("--lora_scale", type=float, default=1.0, help="Scale for LoRA weights.")
160
  parser.add_argument("--output_file", type=str, default="output.png", help="Output file name for the generated image.")
161
  parser.add_argument("--num_inference_steps", type=int, default=30, help="Number of inference steps.")
 
155
  def main():
156
  parser = argparse.ArgumentParser(description="Generate images using a custom diffusion pipeline with LoRA weights.")
157
  parser.add_argument("--model_id", type=str, default='black-forest-labs/FLUX.1-dev', help="Model ID from Hugging Face Hub.")
158
+ parser.add_argument("--adapter_id", type=str, default='pytorch_lora_weights.safetensors', help="LoRA weights file.")
159
  parser.add_argument("--lora_scale", type=float, default=1.0, help="Scale for LoRA weights.")
160
  parser.add_argument("--output_file", type=str, default="output.png", help="Output file name for the generated image.")
161
  parser.add_argument("--num_inference_steps", type=int, default=30, help="Number of inference steps.")