Spaces:
Runtime error
Runtime error
File size: 1,686 Bytes
2de3774 |
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 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 |
import modules.async_worker as worker
import shared
import glob
from pathlib import Path
import datetime
import re
import json
from huggingface_hub import snapshot_download
from PIL import Image
class pipeline:
pipeline_type = ["hugginface_dl"]
model_hash = ""
def parse_gen_data(self, gen_data):
gen_data["original_image_number"] = gen_data["image_number"]
gen_data["image_number"] = 1
gen_data["show_preview"] = False
return gen_data
def load_base_model(self, name):
# We're not doing models here
return
def load_keywords(self, lora):
filename = lora.replace(".safetensors", ".txt")
try:
with open(filename, "r") as file:
data = file.read()
return data
except FileNotFoundError:
return " "
def load_loras(self, loras):
return
def refresh_controlnet(self, name=None):
return
def clean_prompt_cond_caches(self):
return
def process(
self,
gen_data=None,
callback=None,
):
repo = gen_data["prompt"]
print(f"Downloading: {repo}")
repo_id = re.sub(r"^\s*hf:\s*", "", repo, count=1)
repo_id = re.sub(r"\s.*$", "", repo_id, count=1)
repo_id = repo_id.replace(",", "")
worker.add_result(
gen_data["task_id"],
"preview",
(-1, f"Downloading {repo_id}...", None)
)
snapshot_download(
repo_id=repo_id,
local_dir=f"models/diffusers/{repo_id}",
resume_download=True,
)
images = ["html/logo.png"]
return images
|