Views
No views yet
...
embed_dim = 1024 if args.sd_model == "stabilityai/stable-diffusion-2-1" else 768
adaptive_mapping = AdaptiveTokenMapping_v1(embed_dim, embed_dim, embed_dim).to(dtype=torch.float16)
adaptive_mapping.load_state_dict(torch.load(os.path.join(path/to/download/dir, 'adaptive_mapping.safetensors'), map_location=torch.device('cpu')), strict=True)
pipe = StableDiffusionAdaptiveTokenPipeline.from_pretrained(args.sd_model, adaptive_mapping=adaptive_mapping, torch_dtype=torch.float16).to(device)
pipe.load_textual_inversion(path/to/download/dir, weight_name='learned_embeds.safetensors', token=args.token_name)
...1@inproceedings{hou2025aitti,
2 title={AITTI: Learning Adaptive Inclusive Token for Text-to-Image Generation},
3 author={Hou, Xinyu and Li, Xiaoming and Loy, Chen Change},
4 booktitle={International Journal of Computer Vision (IJCV)},
5 year={2025}
6}