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

fix wrong exit call in examples

parent 0fdb72ac
......@@ -125,7 +125,7 @@
" mgx.save(model, f\"{file}.mxr\", format=\"msgpack\")\n",
" else:\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"
]
},
......
......@@ -28,6 +28,7 @@ from PIL import Image
import migraphx as mgx
import numpy as np
import os
import sys
import torch
import time
from functools import wraps
......@@ -177,7 +178,7 @@ class StableDiffusionMGX():
mgx.save(model, f"{file}.mxr", format="msgpack")
else:
print(f"No {name} model found. Please download it and re-try.")
os.exit(1)
sys.exit(1)
return model
@measure
......
......@@ -30,6 +30,7 @@ from pydub import AudioSegment
import migraphx as mgx
import os
import numpy as np
import sys
import time
from functools import wraps
......@@ -118,7 +119,7 @@ class WhisperMGX():
mgx.save(model, f"{file}.mxr", format="msgpack")
else:
print(f"No {name} model found. Please download it and re-try.")
os.exit(1)
sys.exit(1)
return model
@property
......
......@@ -125,7 +125,7 @@
" mgx.save(model, f\"{file}.mxr\", format=\"msgpack\")\n",
" else:\n",
" print(f\"No {name} model found. Please download it and re-try.\")\n",
" os.exit(1)\n",
" sys.exit(1)\n",
" 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