fix to work with dim>320

This commit is contained in:
Kohya S
2023-03-10 21:10:22 +09:00
parent e355b5e1d3
commit 4ad8e75291
2 changed files with 5 additions and 5 deletions

View File

@@ -1,6 +1,5 @@
# Convert LoRA to different rank approximation (should only be used to go to lower rank)
# This code is based off the extract_lora_from_models.py file which is based on https://github.com/cloneofsimo/lora/blob/develop/lora_diffusion/cli_svd.py
# Thanks to cloneofsimo and kohya
import argparse
import torch