mirror of
https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
synced 2025-04-07 21:39:01 +08:00
149824a515
currently packaging as a nixosModule so should be easy to integrate with nixos generators in the future. Also looking at making a docker image without systemd
53 lines
1.5 KiB
Nix
53 lines
1.5 KiB
Nix
{ pkgs, ... }:
|
|
{
|
|
url,
|
|
name,
|
|
hash ? "",
|
|
sha256 ? "",
|
|
...
|
|
}@args:
|
|
pkgs.fetchurl (
|
|
args
|
|
// {
|
|
name = "model";
|
|
passthru = (
|
|
{
|
|
inherit name;
|
|
}
|
|
// args.passthru
|
|
);
|
|
curlOpts = "-H @/build/ACTIVE_TOKEN";
|
|
netrcImpureEnvVars = [
|
|
"HF_TOKEN"
|
|
"CIVITAI_API_TOKEN"
|
|
];
|
|
netrcPhase = ''
|
|
# Warn if HF_TOKEN or CIVITAI_API_TOKEN are not set or didn't work, in
|
|
# the case of failure to fetch
|
|
warnEmptyTokensHook() {
|
|
if [ -z "$HF_TOKEN" ]; then
|
|
echo "Warning: HF_TOKEN is not set. Please set it to access gated Hugging Face models."
|
|
elif [ -n "$HF_TOKEN" ]; then
|
|
echo "Warning: HF_TOKEN is set, but fetching didn't seem to work, check your token!"
|
|
fi
|
|
if [ -z "$CIVITAI_API_TOKEN" ]; then
|
|
echo "Warning: CIVITAI_API_TOKEN is not set. Please set it to access gated CivitAI resources."
|
|
elif [ -n "$CIVITAI_API_TOKEN" ]; then
|
|
echo "Warning: CIVITAI_API_TOKEN is set, but fetching didn't seem to work, check your token!"
|
|
fi
|
|
}
|
|
failureHooks+=(warnEmptyTokensHook)
|
|
|
|
# echo is a bash internal and doesn't create a process in /proc/*/cmdline
|
|
if [[ '${url}' == *huggingface* ]]; then
|
|
echo "Authorization: Bearer $HF_TOKEN" > /build/ACTIVE_TOKEN
|
|
elif [[ '${url}' == *civitai* ]]; then
|
|
echo "Authorization: Bearer $CIVITAI_API_TOKEN" > /build/ACTIVE_TOKEN
|
|
else
|
|
> /build/ACTIVE_TOKEN
|
|
fi
|
|
'';
|
|
inherit url hash sha256;
|
|
}
|
|
)
|