Upload app.py
Browse files
app.py
CHANGED
|
@@ -18,6 +18,7 @@ fill(out, value)
|
|
| 18 |
pipe = UnCLIPPipeline.from_pretrained("kakaobrain/karlo-v1-alpha", torch_dtype=torch.float16)
|
| 19 |
pipe = pipe.to("cuda")
|
| 20 |
|
|
|
|
| 21 |
def run(prompt):
|
| 22 |
images = pipe(prompt, num_images_per_prompt=6).images
|
| 23 |
return images
|
|
|
|
| 18 |
pipe = UnCLIPPipeline.from_pretrained("kakaobrain/karlo-v1-alpha", torch_dtype=torch.float16)
|
| 19 |
pipe = pipe.to("cuda")
|
| 20 |
|
| 21 |
+
@ti.func
|
| 22 |
def run(prompt):
|
| 23 |
images = pipe(prompt, num_images_per_prompt=6).images
|
| 24 |
return images
|