S mode for OPT
This commit is contained in:
parent
a1c7017ddc
commit
376e76f5da
|
@ -515,7 +515,7 @@ def loadmodelsettings():
|
||||||
js = json.load(open(vars.custmodpth.replace('/', '_') + "/config.json", "r"))
|
js = json.load(open(vars.custmodpth.replace('/', '_') + "/config.json", "r"))
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
js = {}
|
js = {}
|
||||||
if vars.model_type == "xglm" or js.get("compat", "j") == "fairseq_lm":
|
if vars.model_type == "xglm" or vars.model_type == "opt" or js.get("compat", "j") == "fairseq_lm":
|
||||||
vars.newlinemode = "s" # Default to </s> newline mode if using XGLM
|
vars.newlinemode = "s" # Default to </s> newline mode if using XGLM
|
||||||
vars.modelconfig = js
|
vars.modelconfig = js
|
||||||
if("badwordsids" in js):
|
if("badwordsids" in js):
|
||||||
|
|
Loading…
Reference in New Issue