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

Merge pull request #29 from hodanov/feature/repair_bug

Browse Source 
Repair some bugs.
This commit is contained in:
hodanov 2023-09-12 20:45:35 +09:00 committed by GitHub
commit 5306610337
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 3 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,

4
setup_files/__main__.py
View File

@ -1,13 +1,13 @@
from __future__ import annotations from __future__ import annotations
from setup import stub from setup import stub
from txt2img import StableDiffusion, new_stable_diffusion from txt2img import 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__":