From e0b686b475885f0c8480f7173eaa7359adf17e27 Mon Sep 17 00:00:00 2001 From: Volpeon Date: Mon, 26 Dec 2022 14:24:21 +0100 Subject: Set default dimensions to 768; add config inheritance --- common.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) (limited to 'common.py') diff --git a/common.py b/common.py index 7ffa77f..f369475 100644 --- a/common.py +++ b/common.py @@ -1,9 +1,23 @@ from pathlib import Path +import json + import torch from transformers import CLIPTextModel, CLIPTokenizer +def load_config(filename): + with open(filename, 'rt') as f: + config = json.load(f) + + args = config["args"] + + if "base" in config: + args = load_config(Path(filename).parent.joinpath(config["base"])) | args + + return args + + def load_text_embedding(embeddings, token_id, file): data = torch.load(file, map_location="cpu") -- cgit v1.2.3-54-g00ecf