19 lines
355 B
Python
19 lines
355 B
Python
from __future__ import annotations
|
|
|
|
from setup import stub
|
|
from txt2img import StableDiffusion, StableDiffusionInterface
|
|
|
|
|
|
def new_stable_diffusion() -> StableDiffusion:
|
|
return StableDiffusion()
|
|
|
|
|
|
@stub.function(gpu="A10G")
|
|
def main():
|
|
sd = new_stable_diffusion()
|
|
print(isinstance(sd, StableDiffusion))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|