Browse Source
Dev 1.0changes added in the opereator file
Signed-off-by: krishna katyal <krishna.katyal@zilliz.com>
main
krishna katyal
3 years ago
1 changed files with
2 additions and
2 deletions
-
animegan.py
|
|
|
@ -20,8 +20,8 @@ class Animegan(Operator): |
|
|
|
super().__init__() |
|
|
|
if framework == 'pytorch': |
|
|
|
import importlib.util |
|
|
|
path = os.path.join(str(Path("__file__").parent), 'pytorch', 'model.py') |
|
|
|
opname = os.path.basename(str(Path("__file__"))).split('.')[0] |
|
|
|
path = os.path.join(str(Path(__file__).parent), 'pytorch', 'model.py') |
|
|
|
opname = os.path.basename(str(Path(__file__))).split('.')[0] |
|
|
|
spec = importlib.util.spec_from_file_location(opname, path) |
|
|
|
module = importlib.util.module_from_spec(spec) |
|
|
|
spec.loader.exec_module(module) |
|
|
|
|
