mirror of
https://github.com/qurator-spk/eynollah.git
synced 2025-07-15 22:09:59 +02:00
let hybrid textline light model be loaded
This commit is contained in:
parent
29e6ad076f
commit
380f59ad67
1 changed files with 6 additions and 2 deletions
|
@ -665,8 +665,12 @@ class Eynollah:
|
||||||
if model_dir in self.models:
|
if model_dir in self.models:
|
||||||
model = self.models[model_dir]
|
model = self.models[model_dir]
|
||||||
else:
|
else:
|
||||||
|
try:
|
||||||
model = load_model(model_dir, compile=False)
|
model = load_model(model_dir, compile=False)
|
||||||
self.models[model_dir] = model
|
self.models[model_dir] = model
|
||||||
|
except:
|
||||||
|
model = load_model(model_dir , compile=False,custom_objects = {"PatchEncoder": PatchEncoder, "Patches": Patches})
|
||||||
|
self.models[model_dir] = model
|
||||||
|
|
||||||
return model, None
|
return model, None
|
||||||
|
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue