mirror of
https://github.com/nod-ai/AMD-SHARK-Studio.git
synced 2026-04-03 03:00:17 -04:00
21 lines
501 B
Python
21 lines
501 B
Python
from apps.shark_studio.modules.ckpt_processing import save_irpa
|
|
import argparse
|
|
import safetensors
|
|
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument(
|
|
"--input",
|
|
type=str,
|
|
default="",
|
|
help="input safetensors/irpa",
|
|
)
|
|
parser.add_argument(
|
|
"--prefix",
|
|
type=str,
|
|
default="",
|
|
help="prefix to add to all the keys in the irpa",
|
|
)
|
|
args = parser.parse_args()
|
|
output_file = save_irpa(args.input, args.prefix)
|
|
print("saved irpa to", output_file, "with prefix", args.prefix)
|