Commit 6090ad68 authored by Attila Dusnoki's avatar Attila Dusnoki
Browse files

fix wrong exit call in examples

parent 0fdb72ac
...@@ -125,7 +125,7 @@ ...@@ -125,7 +125,7 @@
" mgx.save(model, f\"{file}.mxr\", format=\"msgpack\")\n", " mgx.save(model, f\"{file}.mxr\", format=\"msgpack\")\n",
" else:\n", " else:\n",
" print(f\"No {name} model found. Please verify the path is correct and re-try, or re-download model.\")\n", " print(f\"No {name} model found. Please verify the path is correct and re-try, or re-download model.\")\n",
" os.exit(1)\n", " sys.exit(1)\n",
" return model" " return model"
] ]
}, },
......
...@@ -28,6 +28,7 @@ from PIL import Image ...@@ -28,6 +28,7 @@ from PIL import Image
import migraphx as mgx import migraphx as mgx
import numpy as np import numpy as np
import os import os
import sys
import torch import torch
import time import time
from functools import wraps from functools import wraps
...@@ -177,7 +178,7 @@ class StableDiffusionMGX(): ...@@ -177,7 +178,7 @@ class StableDiffusionMGX():
mgx.save(model, f"{file}.mxr", format="msgpack") mgx.save(model, f"{file}.mxr", format="msgpack")
else: else:
print(f"No {name} model found. Please download it and re-try.") print(f"No {name} model found. Please download it and re-try.")
os.exit(1) sys.exit(1)
return model return model
@measure @measure
......
...@@ -30,6 +30,7 @@ from pydub import AudioSegment ...@@ -30,6 +30,7 @@ from pydub import AudioSegment
import migraphx as mgx import migraphx as mgx
import os import os
import numpy as np import numpy as np
import sys
import time import time
from functools import wraps from functools import wraps
...@@ -118,7 +119,7 @@ class WhisperMGX(): ...@@ -118,7 +119,7 @@ class WhisperMGX():
mgx.save(model, f"{file}.mxr", format="msgpack") mgx.save(model, f"{file}.mxr", format="msgpack")
else: else:
print(f"No {name} model found. Please download it and re-try.") print(f"No {name} model found. Please download it and re-try.")
os.exit(1) sys.exit(1)
return model return model
@property @property
......
...@@ -125,7 +125,7 @@ ...@@ -125,7 +125,7 @@
" mgx.save(model, f\"{file}.mxr\", format=\"msgpack\")\n", " mgx.save(model, f\"{file}.mxr\", format=\"msgpack\")\n",
" else:\n", " else:\n",
" print(f\"No {name} model found. Please download it and re-try.\")\n", " print(f\"No {name} model found. Please download it and re-try.\")\n",
" os.exit(1)\n", " sys.exit(1)\n",
" return model" " return model"
] ]
}, },
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment