m

babe.py

@@ -0,0 +1,24 @@
+""" from main import get_nude
+from PIL import Image
+
+x = get_nude(Image.open("girl.png"))
+x[0].save("babe.png")
+
+x = get_nude(Image.open("girl.png"), loras="breastsizeslideroffset.safetensors", lora_weights="1")
+x[0].save("babe2.png") """
+
+import concurrent.futures
+from main import get_nude
+from PIL import Image
+
+def process_image_1():
+    x = get_nude(Image.open("girl.png"))
+    x[0].save("babe.png")
+
+def process_image_2():
+    x = get_nude(Image.open("girl.png"), loras="breastsizeslideroffset.safetensors", lora_weights="1")
+    x[0].save("babe2.png")
+
+with concurrent.futures.ThreadPoolExecutor() as executor:
+    executor.submit(process_image_1)
+    executor.submit(process_image_2)

checkpoints/groundingdino_swint_ogc.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3b3ca2563c77c69f651d7bd133e97139c186df06231157a64c507099c52bc799
+size 693997677

checkpoints/sam_hq_vit_h.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a7ac14a085326d9fa6199c8c698c4f0e7280afdbb974d2c4660ec60877b45e35
+size 2570940653

config/GroundingDINO_SwinB.cfg.py

@@ -0,0 +1,43 @@
+batch_size = 1
+modelname = "groundingdino"
+backbone = "swin_B_384_22k"
+position_embedding = "sine"
+pe_temperatureH = 20
+pe_temperatureW = 20
+return_interm_indices = [1, 2, 3]
+backbone_freeze_keywords = None
+enc_layers = 6
+dec_layers = 6
+pre_norm = False
+dim_feedforward = 2048
+hidden_dim = 256
+dropout = 0.0
+nheads = 8
+num_queries = 900
+query_dim = 4
+num_patterns = 0
+num_feature_levels = 4
+enc_n_points = 4
+dec_n_points = 4
+two_stage_type = "standard"
+two_stage_bbox_embed_share = False
+two_stage_class_embed_share = False
+transformer_activation = "relu"
+dec_pred_bbox_embed_share = True
+dn_box_noise_scale = 1.0
+dn_label_noise_ratio = 0.5
+dn_label_coef = 1.0
+dn_bbox_coef = 1.0
+embed_init_tgt = True
+dn_labelbook_size = 2000
+max_text_len = 256
+text_encoder_type = "bert-base-uncased"
+use_text_enhancer = True
+use_fusion_layer = True
+use_checkpoint = True
+use_transformer_ckpt = True
+use_text_cross_attention = True
+text_dropout = 0.0
+fusion_dropout = 0.0
+fusion_droppath = 0.1
+sub_sentence_present = True

config/GroundingDINO_SwinT_OGC.py

@@ -0,0 +1,43 @@
+batch_size = 1
+modelname = "groundingdino"
+backbone = "swin_T_224_1k"
+position_embedding = "sine"
+pe_temperatureH = 20
+pe_temperatureW = 20
+return_interm_indices = [1, 2, 3]
+backbone_freeze_keywords = None
+enc_layers = 6
+dec_layers = 6
+pre_norm = False
+dim_feedforward = 2048
+hidden_dim = 256
+dropout = 0.0
+nheads = 8
+num_queries = 900
+query_dim = 4
+num_patterns = 0
+num_feature_levels = 4
+enc_n_points = 4
+dec_n_points = 4
+two_stage_type = "standard"
+two_stage_bbox_embed_share = False
+two_stage_class_embed_share = False
+transformer_activation = "relu"
+dec_pred_bbox_embed_share = True
+dn_box_noise_scale = 1.0
+dn_label_noise_ratio = 0.5
+dn_label_coef = 1.0
+dn_bbox_coef = 1.0
+embed_init_tgt = True
+dn_labelbook_size = 2000
+max_text_len = 256
+text_encoder_type = "bert-base-uncased"
+use_text_enhancer = True
+use_fusion_layer = True
+use_checkpoint = True
+use_transformer_ckpt = True
+use_text_cross_attention = True
+text_dropout = 0.0
+fusion_dropout = 0.0
+fusion_droppath = 0.1
+sub_sentence_present = True

controlnettest.py

@@ -0,0 +1,74 @@
+# !pip install transformers accelerate
+from diffusers import StableDiffusionControlNetInpaintPipeline, ControlNetModel, DDIMScheduler
+from diffusers.utils import load_image
+import numpy as np
+import torch
+from dw_pose.main import dwpose
+from PIL import Image
+
+init_image = load_image(
+    "https://huggingface.co/datasets/diffusers/test-arrays/resolve/main/stable_diffusion_inpaint/boy.png"
+)
+# init_image = init_image.resize((512, 512))
+init_image = Image.open("./girl.png")
+
+generator = torch.Generator(device="cpu").manual_seed(1)
+
+mask_image = load_image(
+    "https://huggingface.co/datasets/diffusers/test-arrays/resolve/main/stable_diffusion_inpaint/boy_mask.png"
+)
+# mask_image = mask_image.resize((512, 512))
+mask_image = Image.open("./girlmask.png")
+
+print("doing dwpose")
+
+control_image = dwpose(init_image, 512)
+
+print("done dwpose")
+
+control_image.save("thecontrolfucking.png")
+
+print("loading openpose")
+
+controlnet = ControlNetModel.from_pretrained(
+    "lllyasviel/control_v11p_sd15_openpose", torch_dtype=torch.float16
+)
+pipe = StableDiffusionControlNetInpaintPipeline.from_pretrained(
+    "runwayml/stable-diffusion-v1-5", controlnet=controlnet, torch_dtype=torch.float16
+)
+
+print("loaded openpsoe")
+
+# speed up diffusion process with faster scheduler and memory optimization
+pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)
+
+pipe.enable_model_cpu_offload()
+
+pipe.safety_checker = None
+
+init_image_width, init_image_height = init_image.size
+
+from main import resize
+
+new_size = resize(
+        init_image_width, init_image_height, 1500)
+
+init_image_resized = init_image.resize((int(new_size['width']), int(new_size['height'])))
+mask_image_resized = mask_image.resize((int(new_size['width']), int(new_size['height'])))
+control_image_resized = control_image.resize((int(new_size['width']), int(new_size['height'])))
+
+print("ccreating image")
+
+# generate image
+image = pipe(
+    "a fucking sexy girl naked",
+    num_inference_steps=20,
+    generator=generator,
+    guidance_scale=6.0,
+    eta=1.0,
+    image=init_image_resized,
+    mask_image=mask_image_resized,
+    control_image=control_image_resized,
+).images[0].save("aha.png")
+
+print("created image and saved")