mirror of
https://github.com/modelscope/DiffSynth-Studio.git
synced 2026-04-24 23:26:15 +00:00
fix version issue of transformers (#1412)
This commit is contained in:
9
diffsynth/utils/state_dict_converters/dino_v3.py
Normal file
9
diffsynth/utils/state_dict_converters/dino_v3.py
Normal file
@@ -0,0 +1,9 @@
|
||||
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
|
||||
Reference in New Issue
Block a user