Mercurial > repos > bgruening > black_forest_labs_flux
comparison flux.py @ 4:9ac73a3ec764 draft default tip
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/flux commit d9ac789235499840b8965cbe5efe1d26406f5d9a
| author | bgruening |
|---|---|
| date | Mon, 02 Dec 2024 20:25:54 +0000 |
| parents | eb01f32e9aee |
| children |
comparison
equal
deleted
inserted
replaced
| 3:d15a4b104995 | 4:9ac73a3ec764 |
|---|---|
| 10 if prompt_type == "file": | 10 if prompt_type == "file": |
| 11 with open(sys.argv[3], "r") as f: | 11 with open(sys.argv[3], "r") as f: |
| 12 prompt = f.read().strip() | 12 prompt = f.read().strip() |
| 13 elif prompt_type == "text": | 13 elif prompt_type == "text": |
| 14 prompt = sys.argv[3] | 14 prompt = sys.argv[3] |
| 15 print(f"Creating image from prompt: {prompt}") | |
| 15 | 16 |
| 16 if "dev" in model_path: | 17 if "dev" in model_path: |
| 17 num_inference_steps = 20 | 18 num_inference_steps = 20 |
| 18 elif "schnell" in model_path: | 19 elif "schnell" in model_path: |
| 19 num_inference_steps = 4 | 20 num_inference_steps = 4 |