Commit 5dcc2260 authored by zhaohu xing's avatar zhaohu xing
Browse files

add hash and fix undo hijack bug 



Signed-off-by: default avatarzhaohu xing <920232796@qq.com>
parent a25dfebe

.DS_Store

0 → 100644
+6 KiB

File added.

No diff preview for this file type.

View file

launch.py

+5 −5
Original line number Diff line number Diff line
@@ -234,11 +234,11 @@ def prepare_enviroment(): 


    os.makedirs(dir_repos, exist_ok=True)



    git_clone(stable_diffusion_repo, repo_dir('stable-diffusion-stability-ai'), "Stable Diffusion", )

    git_clone(taming_transformers_repo, repo_dir('taming-transformers'), "Taming Transformers", )

    git_clone(k_diffusion_repo, repo_dir('k-diffusion'), "K-diffusion", )

    git_clone(codeformer_repo, repo_dir('CodeFormer'), "CodeFormer", )

    git_clone(blip_repo, repo_dir('BLIP'), "BLIP", )

    git_clone(stable_diffusion_repo, repo_dir('stable-diffusion-stability-ai'), "Stable Diffusion", stable_diffusion_commit_hash)

    git_clone(taming_transformers_repo, repo_dir('taming-transformers'), "Taming Transformers", taming_transformers_commit_hash)

    git_clone(k_diffusion_repo, repo_dir('k-diffusion'), "K-diffusion", k_diffusion_commit_hash)

    git_clone(codeformer_repo, repo_dir('CodeFormer'), "CodeFormer", codeformer_commit_hash)

    git_clone(blip_repo, repo_dir('BLIP'), "BLIP", blip_commit_hash)



    if not is_installed("lpips"):

        run_pip(f"install -r {os.path.join(repo_dir('CodeFormer'), 'requirements.txt')}", "requirements for CodeFormer")

modules/sd_hijack.py

+5 −1
Original line number Diff line number Diff line
@@ -112,7 +112,11 @@ class StableDiffusionModelHijack: 
        self.layers = flatten(m)



    def undo_hijack(self, m):

        if type(m.cond_stage_model) == sd_hijack_clip.FrozenCLIPEmbedderWithCustomWords:



        if shared.text_model_name == "XLMR-Large":

            m.cond_stage_model = m.cond_stage_model.wrapped 



        elif type(m.cond_stage_model) == sd_hijack_clip.FrozenCLIPEmbedderWithCustomWords:

            m.cond_stage_model = m.cond_stage_model.wrapped



            model_embeddings = m.cond_stage_model.transformer.text_model.embeddings

v2-inference.yaml→v2-inference-v.yaml

+1 −0
Original line number Diff line number Diff line
@@ -2,6 +2,7 @@ model: 
  base_learning_rate: 1.0e-4

  target: ldm.models.diffusion.ddpm.LatentDiffusion

  params:

    parameterization: "v"

    linear_start: 0.00085

    linear_end: 0.0120

    num_timesteps_cond: 1

CRA Git | Maintained and supported by SUSTech CRA and CCSE