22

app.py

@@ -4,40 +4,19 @@ import os, random, json, spaces, torch, time, subprocess
 
 import torch
 # from transformers import AutoProcessor, AutoTokenizer
-# from longcat_image.models import LongCatImageTransformer2DModel
-# from longcat_image.pipelines import LongCatImagePipeline
+from diffusers import NewbiePipeline
 
 from utils.image_utils import rescale_image
 from utils.prompt_utils import polish_prompt
 
+MODEL_REPO = "NewBie-AI/NewBie-image-Exp0.1"
+MAX_SEED = np.iinfo(np.int32).max
 
-# GIT_DIR = "LongCat-Image"
-# GIT_URL = "https://github.com/yourusername/LongCat-Image.git"
-
-# if not os.path.isdir(GIT_DIR):
-#     subprocess.run(["git", "clone", GIT_URL])
-# else:
-#     print("Folder already exists.")
-# MODEL_REPO = "meituan-longcat/LongCat-Image"
-# MAX_SEED = np.iinfo(np.int32).max
-
-# text_processor = AutoTokenizer.from_pretrained( 
-#     MODEL_REPO, 
-#     subfolder = 'tokenizer'
-# )
-# transformer = LongCatImageTransformer2DModel.from_pretrained( 
-#     MODEL_REPO , 
-#     subfolder = 'transformer', 
-#     torch_dtype=torch.bfloat16, 
-#     use_safetensors=True
-# ).to("cuda")
-
-# pipe = LongCatImagePipeline.from_pretrained(
-#     MODEL_REPO,
-#     transformer=transformer,
-#     text_processor=text_processor
-# )
-# pipe.to("cuda", torch.bfloat16)
+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

requirements.txt

@@ -3,4 +3,4 @@ torch
 transformers
 accelerate
 spaces
-git+https://github.com/huggingface/diffusers.git
+git+https://github.com/E-Anlia/diffusers

utils/repo_sync.py

@@ -0,0 +1,7 @@
+# GIT_DIR = "LongCat-Image"
+# GIT_URL = "https://github.com/yourusername/LongCat-Image.git"
+
+# if not os.path.isdir(GIT_DIR):
+#     subprocess.run(["git", "clone", GIT_URL])
+# else:
+#     print("Folder already exists.")