Merge pull request #29 from hodanov/feature/repair_bug
Repair some bugs.
This commit is contained in:
commit
5306610337
@ -32,7 +32,7 @@ def main(
|
||||
if seed == -1:
|
||||
seed_generated = util.generate_seed()
|
||||
start_time = time.time()
|
||||
images = stub.app.run_inference.call(
|
||||
images = stub.app.run_inference.remote(
|
||||
prompt=prompt,
|
||||
n_prompt=n_prompt,
|
||||
height=height,
|
||||
|
||||
@ -1,13 +1,13 @@
|
||||
from __future__ import annotations
|
||||
|
||||
from setup import stub
|
||||
from txt2img import StableDiffusion, new_stable_diffusion
|
||||
from txt2img import new_stable_diffusion
|
||||
|
||||
|
||||
@stub.function(gpu="A10G")
|
||||
def main():
|
||||
sd = new_stable_diffusion()
|
||||
print(isinstance(sd, StableDiffusion))
|
||||
print(f"Deploy '{sd.__class__.__name__}'.")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user