Files
DiffSynth-Studio/diffsynth/utils/state_dict_converters/dino_v3.py
2026-04-24 17:01:08 +08:00

10 lines
292 B
Python

def DINOv3StateDictConverter(state_dict):
new_state_dict = {}
for key in state_dict:
value = state_dict[key]
if key.startswith("layer"):
new_state_dict["model." + key] = value
else:
new_state_dict[key] = value
return new_state_dict