chunzhi/stable-diffusion-cli-on-modal
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Refactor setup.py.

Browse Source
This commit is contained in:
hodanov
2024-11-02 17:23:41 +09:00
parent 1f2e866971
commit b9b8f20736
4 changed files with 142 additions and 102 deletions
Download Patch File Download Diff File Expand all files Collapse all files
cmd/sd15_img2img.py
+6 -2
View File
@@ -4,7 +4,9 @@ import modal
import util
stub = modal.Stub("run-stable-diffusion-cli")
stub.run_inference = modal.Function.from_name("stable-diffusion-cli", "SD15.run_img2img_inference")
stub.run_inference = modal.Function.from_name(
"stable-diffusion-cli", "SD15.run_img2img_inference"
)
@stub.local_entrypoint()
@@ -44,7 +46,9 @@ def main(
)
util.save_images(directory, images, seed_generated, i, output_format)
total_time = time.time() - start_time
print(f"Sample {i} took {total_time:.3f}s ({(total_time)/len(images):.3f}s / image).")
print(
f"Sample {i} took {total_time:.3f}s ({(total_time)/len(images):.3f}s / image)."
)
prompts: dict[str, int | str] = {
"prompt": prompt,
cmd/sd15_txt2img.py
+6 -2
View File
@@ -4,7 +4,9 @@ import modal
import util
app = modal.App("run-stable-diffusion-cli")
run_inference = modal.Function.from_name("stable-diffusion-cli", "SDXLTxt2Img.run_inference")
run_inference = modal.Function.from_name(
"stable-diffusion-cli", "SD15.run_txt2img_inference"
)
@app.local_entrypoint()
@@ -46,7 +48,9 @@ def main(
)
util.save_images(directory, images, seed_generated, i, output_format)
total_time = time.time() - start_time
print(f"Sample {i} took {total_time:.3f}s ({(total_time)/len(images):.3f}s / image).")
print(
f"Sample {i} took {total_time:.3f}s ({(total_time)/len(images):.3f}s / image)."
)
prompts: dict[str, int | str] = {
"prompt": prompt,