22

app.py

@@ -21,6 +21,7 @@ text_encoder_2 = AutoModel.from_pretrained(model_path, subfolder="text_encoder_2
 pipe = NewbiePipeline.from_pretrained(model_path, text_encoder_2=text_encoder_2, torch_dtype=torch.bfloat16)
 del text_encoder_2
 
+
 # pipe = NewbiePipeline.from_pretrained(
 #     MODEL_REPO,
 #     torch_dtype=torch.bfloat16,
@@ -32,38 +33,38 @@ del text_encoder_2
 #     polished_prompt = polish_prompt(prompt)
 #     return polished_prompt, True
 
-# @spaces.GPU
-# def inference(
-#     prompt,
-#     negative_prompt="blurry ugly bad",
-#     width=1024,
-#     height=1024,
-#     seed=42,
-#     randomize_seed=True,
-#     guidance_scale=1.5,
-#     num_inference_steps=8,
-#     progress=gr.Progress(track_tqdm=True),
-# ):
-#     timestamp = time.time()
-#     print(f"timestamp: {timestamp}")
-
-
-#     # generation
-#     if randomize_seed: seed = random.randint(0, MAX_SEED)
-#     generator = torch.Generator().manual_seed(seed)
-
-#     image = pipe(
-#         prompt= prompt,
-#         negative_prompt = negative_prompt,
-#         width=width,
-#         height=height,
-#         generator=generator,
-#         guidance_scale=guidance_scale,
-#         num_inference_steps=num_inference_steps,
-#         enable_prompt_rewrite= False
-#     ).images[0]
-
-#     return image, seed
+@spaces.GPU
+def inference(
+    prompt,
+    negative_prompt="blurry ugly bad",
+    width=1024,
+    height=1024,
+    seed=42,
+    randomize_seed=True,
+    guidance_scale=1.5,
+    num_inference_steps=8,
+    progress=gr.Progress(track_tqdm=True),
+):
+    timestamp = time.time()
+    print(f"timestamp: {timestamp}")
+
+
+    # generation
+    if randomize_seed: seed = random.randint(0, MAX_SEED)
+    generator = torch.Generator().manual_seed(seed)
+
+    image = pipe(
+        prompt= prompt,
+        negative_prompt = negative_prompt,
+        width=width,
+        height=height,
+        generator=generator,
+        guidance_scale=guidance_scale,
+        num_inference_steps=num_inference_steps,
+        enable_prompt_rewrite= False
+    ).images[0]
+
+    return image, seed
 
 
 def read_file(path: str) -> str:

requirements.txt

@@ -4,4 +4,6 @@ transformers
 accelerate
 spaces
 flash-attn
-git+https://github.com/Disty0/diffusers
+git+https://github.com/Disty0/diffusers
+timm
+torchvision