Views
No views yet
1
2
3import torch
4from compel import Compel, ReturnedEmbeddingsType
5from diffusers import DiffusionPipeline
6
7import random
8
9
10negative_prompt = "cartoon, anime, 3d, painting, b&w, low quality"
11
12
13models=["NYUAD-ComNets/Asian_Female_Profession_Model","NYUAD-ComNets/Black_Female_Profession_Model","NYUAD-ComNets/White_Female_Profession_Model",
14"NYUAD-ComNets/Indian_Female_Profession_Model","NYUAD-ComNets/Latino_Hispanic_Female_Profession_Model","NYUAD-ComNets/Middle_Eastern_Female_Profession_Model",
15"NYUAD-ComNets/Asian_Male_Profession_Model","NYUAD-ComNets/Black_Male_Profession_Model","NYUAD-ComNets/White_Male_Profession_Model",
16"NYUAD-ComNets/Indian_Male_Profession_Model","NYUAD-ComNets/Latino_Hispanic_Male_Profession_Model","NYUAD-ComNets/Middle_Eastern_Male_Profession_Model"]
17
18adapters=["asian_female","black_female","white_female","indian_female","latino_female","middle_east_female",
19"asian_male","black_male","white_male","indian_male","latino_male","middle_east_male"]
20
21pipeline = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", variant="fp16", use_safetensors=True, torch_dtype=torch.float16).to("cuda")
22
23
24for i,j in zip(models,adapters):
25 pipeline.load_lora_weights(i, weight_name="pytorch_lora_weights.safetensors",adapter_name=j)
26
27
28pipeline.set_adapters(random.choice(adapters))
29
30
31compel = Compel(tokenizer=[pipeline.tokenizer, pipeline.tokenizer_2] ,
32 text_encoder=[pipeline.text_encoder, pipeline.text_encoder_2],
33 returned_embeddings_type=ReturnedEmbeddingsType.PENULTIMATE_HIDDEN_STATES_NON_NORMALIZED,
34 requires_pooled=[False, True],truncate_long_prompts=False)
35
36
37conditioning, pooled = compel("a photo of a doctor, looking at the camera, closeup headshot facing forward, ultra quality, sharp focus")
38
39negative_conditioning, negative_pooled = compel(negative_prompt)
40[conditioning, negative_conditioning] = compel.pad_conditioning_tensors_to_same_length([conditioning, negative_conditioning])
41
42image = pipeline(prompt_embeds=conditioning, negative_prompt_embeds=negative_conditioning,
43 pooled_prompt_embeds=pooled, negative_pooled_prompt_embeds=negative_pooled,
44 num_inference_steps=40).images[0]
45
46image.save('/../../x.jpg')
47![]() | ![]() | ![]() |
![]() | ![]() | ![]() |
![]() | ![]() | ![]() |
![]() | ![]() | ![]() |
@article{aldahoul2025ai,
title={AI-generated faces influence gender stereotypes and racial homogenization},
author={AlDahoul, Nouar and Rahwan, Talal and Zaki, Yasir},
journal={Scientific reports},
volume={15},
number={1},
pages={14449},
year={2025},
publisher={Nature Publishing Group UK London}
}
@article{aldahoul2024ai,
title={AI-generated faces free from racial and gender stereotypes},
author={AlDahoul, Nouar and Rahwan, Talal and Zaki, Yasir},
journal={arXiv preprint arXiv:2402.01002},
year={2024}
}
@misc{ComNets,
url={[https://huggingface.co/NYUAD-ComNets/Asian_Male_Profession_Model](https://huggingface.co/NYUAD-ComNets/Asian_Male_Profession_Model)},
title={Asian_Male_Profession_Model},
author={Nouar AlDahoul, Talal Rahwan, Yasir Zaki}
}