github/AMD-SHARK-Studio
1
1
Fork 0
mirror of https://github.com/nod-ai/AMD-SHARK-Studio.git synced 2026-04-03 03:00:17 -04:00
Code Issues Packages Projects Releases Wiki Activity

Migration to AMDShark (#2182)

Browse Source 
Signed-off-by: pdhirajkumarprasad <dhirajp@amd.com>
This commit is contained in:
pdhirajkumarprasad
2025-11-20 12:52:07 +05:30
committed by GitHub
parent dba2c8a567
commit fe03539901
232 changed files with 1719 additions and 1719 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
build_tools/stable_diffusion_testing.py
View File

@@ -5,7 +5,7 @@ from apps.stable_diffusion.src.utils.resources import (
get_json_file,
)
from datetime import datetime as dt
from shark.shark_downloader import download_public_file
from amdshark.amdshark_downloader import download_public_file
from image_comparison import compare_images
import argparse
from glob import glob
@@ -178,7 +178,7 @@ def test_loop(
print("Successfully generated image")
os.makedirs("./test_images/golden/" + model_name, exist_ok=True)
download_public_file(
"gs://shark_tank/testdata/golden/" + model_name,
"gs://amdshark_tank/testdata/golden/" + model_name,
"./test_images/golden/" + model_name,
)
test_file_path = os.path.join(
@@ -239,7 +239,7 @@ def test_loop(
def prepare_artifacts():
gen_path = os.path.join(os.getcwd(), "gen_shark_tank")
gen_path = os.path.join(os.getcwd(), "gen_amdshark_tank")
if not os.path.isdir(gen_path):
os.mkdir(gen_path)
for dirname in os.listdir(os.getcwd()):