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

Repair some bugs.

Browse Source
This commit is contained in:
hodanov 2023-09-12 20:42:40 +09:00
parent 729e64ca8e
commit f73b2505d7
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
sdcli/txt2img.py
View File

@ -32,7 +32,7 @@ def main(
if seed == -1: if seed == -1:
seed_generated = util.generate_seed() seed_generated = util.generate_seed()
start_time = time.time() start_time = time.time()
images = stub.app.run_inference.call( images = stub.app.run_inference.remote(
prompt=prompt, prompt=prompt,
n_prompt=n_prompt, n_prompt=n_prompt,
height=height, height=height,

2
setup_files/__main__.py
View File

@ -7,7 +7,7 @@ from txt2img import StableDiffusion, new_stable_diffusion
@stub.function(gpu="A10G") @stub.function(gpu="A10G")
def main(): def main():
sd = new_stable_diffusion() sd = new_stable_diffusion()
print(isinstance(sd, StableDiffusion)) print(f"Deploy '{sd.__class__.__name__}'.")
if __name__ == "__main__": if __name__ == "__main__":