22

app.py

@@ -4,7 +4,8 @@ import os, random, json, spaces, torch, time, subprocess
 
 import torch
 # from transformers import AutoProcessor, AutoTokenizer
-from diffusers import NewbiePipeline
+from diffusers import DiffusionPipeline
+# from diffusers import NewbiePipeline
 
 from utils.image_utils import rescale_image
 from utils.prompt_utils import polish_prompt
@@ -12,11 +13,11 @@ from utils.prompt_utils import polish_prompt
 MODEL_REPO = "NewBie-AI/NewBie-image-Exp0.1"
 MAX_SEED = np.iinfo(np.int32).max
 
-pipe = NewbiePipeline.from_pretrained(
-    MODEL_REPO,
-    torch_dtype=torch.bfloat16,
-)
-pipe.to("cuda")
+# pipe = NewbiePipeline.from_pretrained(
+#     MODEL_REPO,
+#     torch_dtype=torch.bfloat16,
+# )
+# pipe.to("cuda")
 
 # def prepare(prompt, is_polish_prompt):
 #     if not is_polish_prompt: return prompt, False