lint
This commit is contained in:
parent
1aeb72ba58
commit
19dbc442c6
1 changed files with 0 additions and 1 deletions
|
@ -326,7 +326,6 @@ if __name__ == '__main__':
|
||||||
logger.info(f"Loading base model: {dir_base_model.name}")
|
logger.info(f"Loading base model: {dir_base_model.name}")
|
||||||
hparams = Model.load_hparams(dir_base_model)
|
hparams = Model.load_hparams(dir_base_model)
|
||||||
|
|
||||||
|
|
||||||
with torch.inference_mode():
|
with torch.inference_mode():
|
||||||
try:
|
try:
|
||||||
model_class = Model.from_model_architecture(hparams["architectures"][0])
|
model_class = Model.from_model_architecture(hparams["architectures"][0])
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue