auto-stable-diffusion/push_to_hub.py at main · milo157/auto-stable-diffusion · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
import os
import random

from huggingface_hub import HfApi
from huggingface_hub import create_repo

HF_TOKEN = os.getenv("HUGGINGFACE_TOKEN", "")
OUTPUT_DIR = os.getenv("OUTPUT_DIR", "")
UUID = os.getenv("UUID", "")
concept_name = UUID or f"session-{random.randrange(1, 10000)}"

api = HfApi()
username = api.whoami(token=HF_TOKEN)["name"]

repo_id = f"{username}/{concept_name}"


def bar(prg):
    br = "[1;33mUploading to HuggingFace : " '[0m|' + '█' * prg + ' ' * (25 - prg) + '| ' + str(prg * 4) + "%"
    return br


print("Loading...")

print(bar(1))

create_repo(repo_id, private=True, token=HF_TOKEN)

api.create_commit(repo_id=repo_id, operations=[], commit_message=f"Upload the concept {concept_name} embeds and token", token=HF_TOKEN)

api.upload_folder(folder_path=OUTPUT_DIR + "/feature_extractor", path_in_repo="feature_extractor", repo_id=repo_id, token=HF_TOKEN)

print(bar(8))

api.upload_folder(folder_path=OUTPUT_DIR + "/scheduler", path_in_repo="scheduler", repo_id=repo_id, token=HF_TOKEN)

print(bar(9))

api.upload_folder(folder_path=OUTPUT_DIR + "/text_encoder", path_in_repo="text_encoder", repo_id=repo_id, token=HF_TOKEN)

print(bar(12))

api.upload_folder(folder_path=OUTPUT_DIR + "/tokenizer", path_in_repo="tokenizer", repo_id=repo_id, token=HF_TOKEN)

print(bar(13))

api.upload_folder(folder_path=OUTPUT_DIR + "/unet", path_in_repo="unet", repo_id=repo_id, token=HF_TOKEN)

print(bar(21))

api.upload_folder(folder_path=OUTPUT_DIR + "/vae", path_in_repo="vae", repo_id=repo_id, token=HF_TOKEN)

print(bar(23))

api.upload_file(path_or_fileobj=OUTPUT_DIR + "/model_index.json", path_in_repo="model_index.json", repo_id=repo_id, token=HF_TOKEN)

print(bar(24))

print(f"Your concept was saved successfully. https://huggingface.co/{repo_id}")