Finegrain/refiners
1
0
Fork 0
mirror of https://github.com/finegrain-ai/refiners.git synced 2024-11-12 16:18:22 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update scripts/convert-lora-weights.py

Browse source 
Co-authored-by: Cédric Deltheil <355031+deltheil@users.noreply.github.com>
This commit is contained in:
Benjamin Trom 2023-08-15 18:06:42 +02:00 committed by Benjamin Trom
parent c9fba44f39
commit 2ee094c18c
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
scripts/convert-lora-weights.py
View file

@ -57,7 +57,7 @@ def process(source: str, base_model: str, output_file: str) -> None:
diffusers_to_refiners = create_state_dict_mapping( diffusers_to_refiners = create_state_dict_mapping(
source_model=refiners_model, target_model=diffusers_model, source_args=refiners_args, target_args=diffusers_args source_model=refiners_model, target_model=diffusers_model, source_args=refiners_args, target_args=diffusers_args
) )
assert diffusers_to_refiners is not None assert diffusers_to_refiners is not None, "Model conversion failed"
apply_loras_to_target(module=refiners_model, target=LoraTarget(target), rank=rank, scale=1.0) apply_loras_to_target(module=refiners_model, target=LoraTarget(target), rank=rank, scale=1.0)
for layer in refiners_model.layers(layer_type=Lora): for layer in refiners_model.layers(layer_type=Lora):