Upload 2 files

app.py

@@ -1,6 +1,19 @@
 import gradio as gr
-import torch
 from diffusers import UnCLIPPipeline
+import torch
+import taichi as ti
+
+ti.init(arch=ti.cpu)
+
+@ti.kernel
+def fill(out: ti.types.ndarray(), value: ti.types.ndarray()):
+    for I in ti.grouped(out):
+        out[I] = value[None]
+
+out = torch.empty((10,), dtype=torch.int32)
+value = torch.tensor(2, dtype=torch.int32)  # torch scalar
+fill(out, value)
+
 
 pipe = UnCLIPPipeline.from_pretrained("kakaobrain/karlo-v1-alpha", torch_dtype=torch.float16)
 pipe = pipe.to("cuda")

requirements.txt

@@ -1,5 +1,6 @@
-diffusers==0.11.1
-transformers==4.25.1
-safetensors==0.2.6
-accelerate==0.15.0
-torch==2.0.1
+diffusers
+transformers
+safetensors
+accelerate
+torch
+taichi