mirror of
https://github.com/qurator-spk/eynollah.git
synced 2025-06-09 12:19:54 +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:
|
||||
model = self.models[model_dir]
|
||||
else:
|
||||
model = load_model(model_dir, compile=False)
|
||||
self.models[model_dir] = model
|
||||
try:
|
||||
model = load_model(model_dir, compile=False)
|
||||
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
|
||||
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue