Mirrors
/
stable-diffusion-paperspace
mirror of https://github.com/Engineer-of-Stuff/stable-diffusion-paperspace.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

test

This commit is contained in:
frostydad 2022-09-22 19:42:34 -06:00
parent 7f825d8011
commit 127697c3f7
2 changed files with 14 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
.github/workflows/generate-webpage.yml vendored
View File

@ -51,17 +51,13 @@ jobs:
file.close()
return (size, None)
# Get list of model directories
# working_dir = os.getcwd()
# root_dir_list = []
# for x in os.listdir(working_dir):
# if os.path.isdir(x):
# root_dir_list.append(x)
# Get list of models under the sd-concepts-library organization
api = HfApi()
root_dir_list = api.list_models(author="sd-concepts-library").sort()
models_list = []
for model in api.list_models(author="sd-concepts-library"):
models_list.append(model.modelId.replace('sd-concepts-library/', ''))
models_list.sort()
print(models_list)
html_struct = """<!DOCTYPE html>
<html lang="en">
@ -111,8 +107,8 @@ jobs:
# Move the model out
i = 1
for model_name in root_dir_list:
print(f'{i}/{len(root_dir_list)} -> {model_name}')
for model_name in models_list:
print(f'{i}/{len(models_list)} -> {model_name}')
# if os.path.exists(f'{model_name}/learned_embeds.bin'): # double check the file exists since sometimes it hasn't been uploaded yet
# shutil.move(f'{model_name}/learned_embeds.bin', f'{model_name}/{model_name}.pt')
# pass

18
other/generate_tx_models_html.py
View File

@ -26,17 +26,13 @@ def getsizes(uri):
file.close()
return (size, None)
# Get list of model directories
# working_dir = os.getcwd()
# root_dir_list = []
# for x in os.listdir(working_dir):
# if os.path.isdir(x):
# root_dir_list.append(x)
# Get list of models under the sd-concepts-library organization
api = HfApi()
root_dir_list = api.list_models(author="sd-concepts-library").sort()
models_list = []
for model in api.list_models(author="sd-concepts-library"):
models_list.append(model.modelId.replace('sd-concepts-library/', ''))
models_list.sort()
print(models_list)
html_struct = """<!DOCTYPE html>
<html lang="en">
@ -86,8 +82,8 @@ html_struct = """<!DOCTYPE html>
# Move the model out
i = 1
for model_name in root_dir_list:
print(f'{i}/{len(root_dir_list)} -> {model_name}')
for model_name in models_list:
print(f'{i}/{len(models_list)} -> {model_name}')
# if os.path.exists(f'{model_name}/learned_embeds.bin'): # double check the file exists since sometimes it hasn't been uploaded yet
# shutil.move(f'{model_name}/learned_embeds.bin', f'{model_name}/{model_name}.pt')
# pass