22

app.py

@@ -21,6 +21,14 @@ from utils.prompt_utils import polish_prompt
 
 MAX_SEED = np.iinfo(np.int32).max
 
+pipe = LongCatImagePipeline.from_pretrained(
+    "meituan-longcat/LongCat-Image",
+    torch_dtype=torch.bfloat16,
+    low_cpu_mem_usage=False,
+)
+pipe.to("cuda")
+
+
 def prepare(prompt, is_polish_prompt):
     if not is_polish_prompt: return prompt, False
     polished_prompt = polish_prompt(prompt)