-
Notifications
You must be signed in to change notification settings - Fork 54
Open
Description
# Load transformer
transformer = HiDreamImageTransformer2DModel.from_pretrained(HIDREAM_I1_PATH, subfolder="transformer")
transformer.max_seq = 8192
src_dict = transformer.state_dict()
edit_dict = load_safetensors(HIDREAM_E1_PATH + "/transformer")
reload_keys = {"editing": src_dict, "refine": edit_dict}
transformer.load_state_dict(edit_dict, strict=True)
The code is from pipeline_hidream_image_editing.py.
In my opinion, HIDREAM_I1_PATH is for refine, and HIDREAM_E1_PATH is for edit. Howerver, reload_keys = {"editing": src_dict, "refine": edit_dict}
huanpengchu and behindthepixels
Metadata
Metadata
Assignees
Labels
No labels