Skip to content

Commit

Permalink
using deprecated naming + removed extra streamlit entries
Browse files Browse the repository at this point in the history
  • Loading branch information
mmartial committed Jun 7, 2024
1 parent 317ab88 commit b0e6352
Showing 1 changed file with 2 additions and 4 deletions.
6 changes: 2 additions & 4 deletions OpenAI_DallE.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,6 @@
import pathlib

import common_functions as cf
import common_functions_WUI as cfw

from datetime import datetime

Expand Down Expand Up @@ -82,8 +81,8 @@ def set_parameters(self, models_list, av_models_list):
for t_model in s_models_list:
model = t_model.strip()
if model in av_models_list:
if av_models_list[model]["status"] == "retired":
warning += f"Model [{model}] is retired (" + av_models_list[model]['status_details'] + "), discarding it"
if av_models_list[model]["status"] == "deprecated":
warning += f"Model [{model}] is deprecated (" + av_models_list[model]['status_details'] + "), discarding it"
else:
models[model] = dict(av_models_list[model])
if cf.isNotBlank(models[model]["status_details"]):
Expand Down Expand Up @@ -162,7 +161,6 @@ def dalle_it(self, model, prompt, img_size, img_count, dest_dir, st_placeholder
if st_placeholder:
st_placeholder.empty()

runid = cfw.get_runid()
run_file = f"{dest_dir}/run.json"
run_json = {
"prompt": prompt,
Expand Down

0 comments on commit b0e6352

Please sign in to comment.