Skip to content

Commit

Permalink
Merge branch 'latest' into ea/svd
Browse files Browse the repository at this point in the history
  • Loading branch information
eaidova committed Apr 19, 2024
2 parents 20979fa + 0382dbe commit 2f19ebd
Show file tree
Hide file tree
Showing 68 changed files with 4,588 additions and 1,668 deletions.
17 changes: 17 additions & 0 deletions .ci/heavy_ubuntu_gpu.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,17 @@
blip-diffusion-subject-generation
controlnet-stable-diffusion
decidiffusion-image-generation
distil-whisper-asr
film-slowmo
instant-id
instruct-pix2pix-image-editing
knowledge-graphs-conve
llava-multimodal-chatbot
mms-massively-multilingual-speech
paint-by-example
qrcode-monster
softvc-voice-conversion
speech-recognition-quantization
stable-diffusion-v2-text-to-image
whisper-subtitles-generation
zeroscope-text2video
12 changes: 12 additions & 0 deletions .ci/heavy_win_gpu.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
controlnet-stable-diffusion
detectron2-to-openvino
film-slowmo
instant-id
instruct-pix2pix-image-editing
knowledge-graphs-conve
mms-massively-multilingual-speech
paint-by-example
qrcode-monster
sdxl-turbo
stable-diffusion-text-to-image
wuerstchen-image-generation
3 changes: 2 additions & 1 deletion .ci/ignore_convert_execution.txt
Original file line number Diff line number Diff line change
Expand Up @@ -63,4 +63,5 @@ notebooks/llm-question-answering/llm-question-answering.ipynb
notebooks/instant-id/instant-id.ipynb
notebooks/style-transfer-webcam/style-transfer.ipynb
notebooks/llava-next-multimodal-chatbot/llava-next-multimodal-chatbot.ipynb
notebooks/stable-video-diffusion/stable-video-diffusion.ipynb
notebooks/stable-video-diffusion/stable-video-diffusion.ipynb
notebooks/llm-agent-langchain/llm-agent-langchain.ipynb
3 changes: 2 additions & 1 deletion .ci/ignore_treon_docker.txt
Original file line number Diff line number Diff line change
Expand Up @@ -61,4 +61,5 @@ style-transfer-webcam
animate-anyone
llava-next-multimodal-chatbot
llm-rag-langchain
stable-video-diffusion
stable-video-diffusion
llm-agent-langchain
4 changes: 3 additions & 1 deletion .ci/ignore_treon_linux.txt
Original file line number Diff line number Diff line change
Expand Up @@ -52,6 +52,7 @@ stable-zephyr-3b-chatbot
llm-question-answering
stable-diffusion-torchdynamo-backend
stable-diffusion-ip-adapter
stable-diffusion-keras-cv
kosmos2-multimodal-large-language-model
photo-maker
openvoice
Expand All @@ -60,4 +61,5 @@ style-transfer-webcam
animate-anyone
llava-next-multimodal-chatbot
llm-rag-langchain
stable-video-diffusion
stable-video-diffusion
llm-agent-langchain
2 changes: 2 additions & 0 deletions .ci/ignore_treon_mac.txt
Original file line number Diff line number Diff line change
Expand Up @@ -50,6 +50,7 @@ stable-zephyr-3b-chatbot
llm-question-answering
stable-diffusion-torchdynamo-backend
stable-diffusion-ip-adapter
stable-diffusion-keras-cv
mobilevlm-language-assistant
kosmos2-multimodal-large-language-model
photo-maker
Expand All @@ -62,3 +63,4 @@ animate-anyone
llava-next-multimodal-chatbot
llm-rag-langchain
stable-video-diffusion
llm-agent-langchain
4 changes: 3 additions & 1 deletion .ci/ignore_treon_win.txt
Original file line number Diff line number Diff line change
Expand Up @@ -45,6 +45,7 @@ distil-whisper-asr
film-slowmo
sound-generation-audioldm2
sdxl-turbo
stable-diffusion-keras-cv
paint-by-example
stable-zephyr-3b-chatbot
llm-question-answering
Expand All @@ -57,4 +58,5 @@ instant-id
animate-anyone
llava-next-multimodal-chatbot
llm-rag-langchain
stable-video-diffusion
stable-video-diffusion
llm-agent-langchain
1 change: 1 addition & 0 deletions .ci/spellcheck/.pyspelling.wordlist.txt
Original file line number Diff line number Diff line change
Expand Up @@ -603,6 +603,7 @@ Raj
Ranftl
RASPP
rcnn
ReAct
RealSense
RealSR
Realtime
Expand Down
25 changes: 21 additions & 4 deletions .ci/validate_notebooks.py
Original file line number Diff line number Diff line change
Expand Up @@ -41,6 +41,19 @@ def move_notebooks(nb_dir):
shutil.copytree(current_notebooks_dir, nb_dir)


def get_notebooks_subdir(changed_path, orig_nb_dir):
if (orig_nb_dir / changed_path).exists() and (orig_nb_dir / changed_path).is_dir():
notebook_subdir = orig_nb_dir / changed_path
if not list(notebook_subdir.rglob("**/*.ipynb")):
notebook_subdir = None
else:
notebook_subdir = notebook_subdir.relative_to(orig_nb_dir)
print(notebook_subdir)
else:
notebook_subdir = find_notebook_dir(changed_path.resolve(), orig_nb_dir.resolve())
return notebook_subdir


def prepare_test_plan(test_list, ignore_list, nb_dir=None):
orig_nb_dir = ROOT / "notebooks"
notebooks_dir = orig_nb_dir if nb_dir is None else nb_dir
Expand All @@ -63,6 +76,7 @@ def prepare_test_plan(test_list, ignore_list, nb_dir=None):
ignored_notebooks.append(ig_nb)
print(f"ignored notebooks: {ignored_notebooks}")

testing_notebooks = []
if len(test_list) == 1 and test_list[0].endswith(".txt"):
testing_notebooks = []
with open(test_list[0], "r") as f:
Expand All @@ -74,16 +88,19 @@ def prepare_test_plan(test_list, ignore_list, nb_dir=None):
break
if changed_path.suffix == ".md":
continue
notebook_subdir = find_notebook_dir(changed_path.resolve(), orig_nb_dir.resolve())
notebook_subdir = get_notebooks_subdir(changed_path, orig_nb_dir)
if notebook_subdir is None:
continue
testing_notebooks.append(notebook_subdir)
test_list = set(testing_notebooks)
else:
test_list = set(map(lambda x: Path(x), test_list))
for test_item in test_list:
notebook_subdir = get_notebooks_subdir(Path(test_item), orig_nb_dir)
if notebook_subdir is not None:
testing_notebooks.append(notebook_subdir)
test_list = set(testing_notebooks)
print(f"test notebooks: {test_list}")

ignore_list = set(map(lambda x: Path(x), ignored_notebooks))

for notebook in statuses:
if notebook not in test_list:
statuses[notebook]["status"] = "SKIPPED"
Expand Down
Loading

0 comments on commit 2f19ebd

Please sign in to comment.