Flux-Character-Consistancy

Running on Zero

zs38 commited on Jan 20

Commit

83cf50d

1 Parent(s): f763f30

update

Files changed (1) hide show

app.py CHANGED Viewed

@@ -46,7 +46,7 @@ login(token=os.environ['HF_TOKEN'])
 file_path = hf_hub_download(repo_id=adapter_repo_id, filename=adapter_path)
 logger.info("load adapter")
-state_dict = safetensors.torch.load_file(adapter_path)
 state_dict = {'.'.join(k.split('.')[1:]): state_dict[k] for k in state_dict.keys()}
 diff = model.load_state_dict(state_dict, strict=False)
 diff = extra_embedder.load_state_dict(state_dict, strict=False)

 file_path = hf_hub_download(repo_id=adapter_repo_id, filename=adapter_path)
 logger.info("load adapter")
+state_dict = safetensors.torch.load_file(file_path)
 state_dict = {'.'.join(k.split('.')[1:]): state_dict[k] for k in state_dict.keys()}
 diff = model.load_state_dict(state_dict, strict=False)
 diff = extra_embedder.load_state_dict(state_dict, strict=False)