PseudoTerminal X
commited on
Commit
•
a5590b5
1
Parent(s):
17b9398
Update README.md
Browse files
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,
|
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.")
|