safree key error

[ShristiDasBiswas]

Hi, I get the following issue with bash scripts/run_nudity_sdxl.sh .

Traceback (most recent call last):
  File "/home/min/a/sdasbisw/Desktop/PROJECTS/svd_diff/SAFREE-main/generate_safree.py", line 426, in <module>
    main()
  File "/home/min/a/sdasbisw/Desktop/PROJECTS/svd_diff/SAFREE-main/generate_safree.py", line 297, in main
    imgs = pipe(
  File "/home/min/a/sdasbisw/anaconda3/envs/ucd/lib/python3.9/site-packages/torch/utils/_contextlib.py", line 115, in decorate_context
    return func(*args, **kwargs)
  File "/home/min/a/sdasbisw/Desktop/PROJECTS/svd_diff/SAFREE-main/models/modified_stable_diffusion_xl_pipeline.py", line 1601, in __call__
    _text_embeddings = rescaled_text_embeddings if (sf["safree"] \
KeyError: 'safree'

could you help me solve this?

[Intelligent114]

modify generate_safree.py at line 293 as below:

        if 'xl' in args.model_id:
            imgs = pipe(
                target_prompt,
                num_images_per_prompt=args.num_samples,
                guidance_scale=guidance,
                num_inference_steps=args.num_inference_steps,
                negative_prompt=negative_prompt,
                negative_prompt_space=negative_prompt_space,
                height=args.image_length,
                width=args.image_length,
                generator=gen.manual_seed(seed),
                safree=args.safree,
                safree_dict={"re_attn_t": [int(tr) for tr in args.re_attn_t.split(",")],
                                "alpha": args.sf_alpha,
                                "svf": args.self_validation_filter,
                                "logger": logger,
                                "up_t": args.up_t,
                                "category": args.category
                                },    
            ).images
        else:
            imgs = pipe(
                target_prompt,
                num_images_per_prompt=args.num_samples,
                guidance_scale=guidance,
                num_inference_steps=args.num_inference_steps,
                negative_prompt=negative_prompt,
                negative_prompt_space=negative_prompt_space,
                height=args.image_length,
                width=args.image_length,
                generator=gen.manual_seed(seed),
                safree_dict={"re_attn_t": [int(tr) for tr in args.re_attn_t.split(",")],
                                "alpha": args.sf_alpha,
                                "logger": logger,
                                "safree": args.safree,
                                "svf": args.self_validation_filter,
                                "lra": args.latent_re_attention,
                                "up_t": args.up_t,
                                "category": args.category
                                },                
                **(safe_config or {})        
            )

add "safree": args.safree, like below

        if 'xl' in args.model_id:
            imgs = pipe(
                target_prompt,
                num_images_per_prompt=args.num_samples,
                guidance_scale=guidance,
                num_inference_steps=args.num_inference_steps,
                negative_prompt=negative_prompt,
                negative_prompt_space=negative_prompt_space,
                height=args.image_length,
                width=args.image_length,
                generator=gen.manual_seed(seed),
                safree=args.safree,
                safree_dict={"re_attn_t": [int(tr) for tr in args.re_attn_t.split(",")],
                                "alpha": args.sf_alpha,
                                "svf": args.self_validation_filter,
                                "logger": logger,
                                "up_t": args.up_t,
                                "category": args.category,
                                "safree": args.safree,   
                                },    
            ).images
        else:
            imgs = pipe(
                target_prompt,
                num_images_per_prompt=args.num_samples,
                guidance_scale=guidance,
                num_inference_steps=args.num_inference_steps,
                negative_prompt=negative_prompt,
                negative_prompt_space=negative_prompt_space,
                height=args.image_length,
                width=args.image_length,
                generator=gen.manual_seed(seed),
                safree_dict={"re_attn_t": [int(tr) for tr in args.re_attn_t.split(",")],
                                "alpha": args.sf_alpha,
                                "logger": logger,
                                "safree": args.safree,
                                "svf": args.self_validation_filter,
                                "lra": args.latent_re_attention,
                                "up_t": args.up_t,
                                "category": args.category
                                },                
                **(safe_config or {})        
            )