feat(lama-cleaner) add lama-cleaner (#5555)
* feat(lama-cleaner) add lama-cleaner * add icon * add extraArgs * add extraArgs * add host command * set as command * args * no point to test gpu on CI Co-authored-by: Stavros kois <s.kois@outlook.com>
This commit is contained in:
Xstar97
GitHub
parent
8737d5ff2b
commit
82e2285950
|
|
@ -0,0 +1,30 @@
|
|||
# Patterns to ignore when building packages.
|
||||
# This supports shell glob matching, relative path matching, and
|
||||
# negation (prefixed with !). Only one pattern per line.
|
||||
.DS_Store
|
||||
# Common VCS dirs
|
||||
.git/
|
||||
.gitignore
|
||||
.bzr/
|
||||
.bzrignore
|
||||
.hg/
|
||||
.hgignore
|
||||
.svn/
|
||||
# Common backup files
|
||||
*.swp
|
||||
*.bak
|
||||
*.tmp
|
||||
*~
|
||||
# Various IDEs
|
||||
.project
|
||||
.idea/
|
||||
*.tmproj
|
||||
.vscode/
|
||||
# OWNERS file for Kubernetes
|
||||
OWNERS
|
||||
# helm-docs templates
|
||||
*.gotmpl
|
||||
# docs folder
|
||||
/docs
|
||||
# icon
|
||||
icon.png
|
||||
|
|
@ -0,0 +1 @@
|
|||
# Changelog
|
||||
|
|
@ -0,0 +1,31 @@
|
|||
apiVersion: v2
|
||||
appVersion: "0.26.1"
|
||||
dependencies:
|
||||
- name: common
|
||||
repository: https://library-charts.truecharts.org
|
||||
version: 11.0.7
|
||||
deprecated: false
|
||||
description: Open-source inpainting tool powered by SOTA AI model.
|
||||
home: https://truecharts.org/docs/charts/lama-cleaner
|
||||
icon: https://truecharts.org/img/hotlink-ok/chart-icons/lama-cleaner.png
|
||||
keywords:
|
||||
- lama-cleaner
|
||||
- media
|
||||
- AI
|
||||
kubeVersion: ">=1.16.0-0"
|
||||
maintainers:
|
||||
- email: info@truecharts.org
|
||||
name: TrueCharts
|
||||
url: https://truecharts.org
|
||||
name: lama-cleaner
|
||||
sources:
|
||||
- https://github.com/truecharts/charts/tree/master/charts/lama-cleaner
|
||||
- https://github.com/Sanster/lama-cleaner
|
||||
type: application
|
||||
version: 0.0.1
|
||||
annotations:
|
||||
truecharts.org/catagories: |
|
||||
- media
|
||||
- AI
|
||||
truecharts.org/SCALE-support: "true"
|
||||
truecharts.org/grade: U
|
||||
|
|
@ -0,0 +1 @@
|
|||
# lama-cleaner
|
||||
Binary file not shown.
|
After Width: | Height: | Size: 15 KiB |
|
|
@ -0,0 +1,199 @@
|
|||
# Include{groups}
|
||||
portals:
|
||||
open:
|
||||
# Include{portalLink}
|
||||
questions:
|
||||
# Include{global}
|
||||
- variable: imageSelector
|
||||
group: Container Image
|
||||
label: Select Image
|
||||
schema:
|
||||
type: string
|
||||
default: image
|
||||
enum:
|
||||
- value: image
|
||||
description: CPU Image
|
||||
- value: imageGPU
|
||||
description: GPU Image
|
||||
# Include{controller}
|
||||
# Include{replicas}
|
||||
# Include{replica1}
|
||||
# Include{controllerExpertExtraArgs}
|
||||
- variable: lamacleaner
|
||||
group: App Configuration
|
||||
label: Lama Cleaner Configuration
|
||||
schema:
|
||||
additional_attrs: true
|
||||
type: dict
|
||||
attrs:
|
||||
- variable: model
|
||||
label: Model
|
||||
description: Inpainting Model https://github.com/Sanster/lama-cleaner#inpainting-model
|
||||
schema:
|
||||
type: string
|
||||
required: true
|
||||
default: lama
|
||||
enum:
|
||||
- value: lama
|
||||
description: lama
|
||||
- value: cv2
|
||||
description: cv2
|
||||
- value: ldm
|
||||
description: ldm
|
||||
- value: zits
|
||||
description: zits
|
||||
- value: fcf
|
||||
description: fcf
|
||||
- value: sd1.5
|
||||
description: sd1.5
|
||||
- value: mat
|
||||
description: mat
|
||||
- variable: input
|
||||
label: Input
|
||||
description: Path to image you want to load by default
|
||||
schema:
|
||||
type: string
|
||||
default: ""
|
||||
- variable: sd_run_local
|
||||
label: Stable Diffusion Run Local
|
||||
description: Once the model as downloaded, you can enable this.
|
||||
schema:
|
||||
type: boolean
|
||||
default: false
|
||||
- variable: hf_access_token
|
||||
label: HuggingFace Access Token
|
||||
description: Stable diffusion need HuggingFace access token to download model
|
||||
schema:
|
||||
type: string
|
||||
show_if: [[ "sd_run_local", "=", "true"]]
|
||||
private: true
|
||||
default: ""
|
||||
- variable: sd_disable_nsfw
|
||||
label: Stable Diffusion Disable NSFW Checker.
|
||||
description: Disable stable diffusion NSFW checker.
|
||||
schema:
|
||||
type: boolean
|
||||
default: false
|
||||
- variable: sd_cpu_text_encoder
|
||||
label: Stable Diffusion Text Encoder
|
||||
description: Always run stable-diffusion TextEncoder model on CPU.
|
||||
schema:
|
||||
type: boolean
|
||||
default: false
|
||||
- variable: sd_enable_xformers
|
||||
label: Stable Diffusion Enable XFormers
|
||||
description: Enable xFormers optimizations.
|
||||
schema:
|
||||
type: boolean
|
||||
default: false
|
||||
- variable: debug
|
||||
label: Enable debug
|
||||
description: Enable debug mode for flask web server
|
||||
schema:
|
||||
type: boolean
|
||||
default: false
|
||||
# Include{containerConfig}
|
||||
# Include{serviceRoot}
|
||||
- variable: main
|
||||
label: Main Service
|
||||
description: The Primary service on which the healthcheck runs, often the webUI
|
||||
schema:
|
||||
additional_attrs: true
|
||||
type: dict
|
||||
attrs:
|
||||
# Include{serviceSelectorLoadBalancer}
|
||||
# Include{serviceSelectorExtras}
|
||||
- variable: main
|
||||
label: Main Service Port Configuration
|
||||
schema:
|
||||
additional_attrs: true
|
||||
type: dict
|
||||
attrs:
|
||||
- variable: port
|
||||
label: Port
|
||||
description: This port exposes the container port on the service
|
||||
schema:
|
||||
type: int
|
||||
default: 10604
|
||||
required: true
|
||||
# Include{serviceExpertRoot}
|
||||
default: false
|
||||
# Include{serviceExpert}
|
||||
# Include{serviceList}
|
||||
# Include{persistenceRoot}
|
||||
- variable: torch
|
||||
label: App Torch Storage
|
||||
description: Stores the Application Torch.
|
||||
schema:
|
||||
additional_attrs: true
|
||||
type: dict
|
||||
attrs:
|
||||
# Include{persistenceBasic}
|
||||
- variable: huggingface
|
||||
label: App HuggingFace Storage
|
||||
description: Stores the Application HuggingFace.
|
||||
schema:
|
||||
additional_attrs: true
|
||||
type: dict
|
||||
attrs:
|
||||
# Include{persistenceBasic}
|
||||
# Include{persistenceList}
|
||||
# Include{ingressRoot}
|
||||
- variable: main
|
||||
label: Main Ingress
|
||||
schema:
|
||||
additional_attrs: true
|
||||
type: dict
|
||||
attrs:
|
||||
# Include{ingressDefault}
|
||||
# Include{ingressTLS}
|
||||
# Include{ingressTraefik}
|
||||
# Include{ingressList}
|
||||
# Include{security}
|
||||
# Include{securityContextAdvancedRoot}
|
||||
- variable: privileged
|
||||
label: Privileged mode
|
||||
schema:
|
||||
type: boolean
|
||||
default: false
|
||||
- variable: readOnlyRootFilesystem
|
||||
label: ReadOnly Root Filesystem
|
||||
schema:
|
||||
type: boolean
|
||||
default: false
|
||||
- variable: allowPrivilegeEscalation
|
||||
label: Allow Privilege Escalation
|
||||
schema:
|
||||
type: boolean
|
||||
default: false
|
||||
- variable: runAsNonRoot
|
||||
label: runAsNonRoot
|
||||
schema:
|
||||
type: boolean
|
||||
default: false
|
||||
# Include{podSecurityContextRoot}
|
||||
- variable: runAsUser
|
||||
label: runAsUser
|
||||
description: The UserID of the user running the application
|
||||
schema:
|
||||
type: int
|
||||
default: 0
|
||||
- variable: runAsGroup
|
||||
label: runAsGroup
|
||||
description: The groupID this App of the user running the application
|
||||
schema:
|
||||
type: int
|
||||
default: 0
|
||||
- variable: fsGroup
|
||||
label: fsGroup
|
||||
description: The group that should own ALL storage.
|
||||
schema:
|
||||
type: int
|
||||
default: 568
|
||||
# Include{podSecurityContextAdvanced}
|
||||
# Include{resources}
|
||||
# Include{advanced}
|
||||
# Include{addons}
|
||||
# Include{codeserver}
|
||||
# Include{vpn}
|
||||
# Include{documentation}
|
||||
|
|
@ -0,0 +1,33 @@
|
|||
{{- define "lama.args" -}}
|
||||
args:
|
||||
- --host=0.0.0.0
|
||||
- --port={{ .Values.service.main.ports.main.port }}
|
||||
{{- if eq .Values.imageSelector "image" }}
|
||||
- --device=cpu
|
||||
{{- if .Values.lamacleaner.sd_cpu_text_encoder }}
|
||||
- --sd-cpu-textencoder
|
||||
{{- end }}
|
||||
{{- else if eq .Values.imageSelector "gpuImage" }}
|
||||
- --device=cuda
|
||||
{{- end }}
|
||||
{{- with .Values.lamacleaner.model }}
|
||||
- --model={{ . }}
|
||||
{{- end }}
|
||||
{{- if .Values.lamacleaner.sd_run_local }}
|
||||
- --sd-run-local
|
||||
{{- else if .Values.lamacleaner.hf_access_token }}
|
||||
- --hf_access_token={{ .Values.lamacleaner.hf_access_token }}
|
||||
{{- end }}
|
||||
{{- if .Values.lamacleaner.sd_enable_xformers }}
|
||||
- --sd-enable-xformers
|
||||
{{- end }}
|
||||
{{- if .Values.lamacleaner.sd_disable_nsfw }}
|
||||
- --sd-disable-nsfw
|
||||
{{- end }}
|
||||
{{- with .Values.lamacleaner.input }}
|
||||
- --input={{ . }}
|
||||
{{- end }}
|
||||
{{- if .Values.lamacleaner.debug }}
|
||||
- --debug
|
||||
{{- end }}
|
||||
{{- end -}}
|
||||
|
|
@ -0,0 +1,9 @@
|
|||
{{/* Make sure all variables are set properly */}}
|
||||
{{- include "tc.common.loader.init" . }}
|
||||
|
||||
{{- $newArgs := (include "lama.args" . | fromYaml) }}
|
||||
{{- $args := concat .Values.args $newArgs.args }}
|
||||
{{- $_ := set .Values "args" $args -}}
|
||||
|
||||
{{/* Render the templates */}}
|
||||
{{ include "tc.common.loader.apply" . }}
|
||||
|
|
@ -0,0 +1,49 @@
|
|||
image:
|
||||
repository: cwq1913/lama-cleaner
|
||||
pullPolicy: IfNotPresent
|
||||
tag: cpu-0.26.1@sha256:e57095636fe19e62280e78f10f98fff0fe1e0608e689515b2a0a56a7d2dad2e8
|
||||
|
||||
gpuImage:
|
||||
repository: cwq1913/lama-cleaner
|
||||
tag: gpu-0.26.1@sha256:65c54a6309f46bb643be31d704e4b361988b277f7b9492a9dd8e24c28220005c
|
||||
pullPolicy: Always
|
||||
|
||||
imageSelector: image
|
||||
|
||||
command: lama-cleaner
|
||||
|
||||
securityContext:
|
||||
runAsNonRoot: false
|
||||
readOnlyRootFilesystem: false
|
||||
|
||||
podSecurityContext:
|
||||
runAsUser: 0
|
||||
runAsGroup: 0
|
||||
|
||||
lamacleaner:
|
||||
model: lama
|
||||
input: ""
|
||||
sd_run_local: false
|
||||
hf_access_token: ""
|
||||
sd_disable_nsfw: false
|
||||
sd_cpu_text_encoder: false
|
||||
sd_enable_xformers: false
|
||||
debug: false
|
||||
|
||||
service:
|
||||
main:
|
||||
ports:
|
||||
main:
|
||||
protocol: HTTP
|
||||
port: 10604
|
||||
|
||||
persistence:
|
||||
torch:
|
||||
enabled: true
|
||||
mountPath: /root/.cache/torch
|
||||
huggingface:
|
||||
enabled: true
|
||||
mountPath: /root/.cache/huggingface
|
||||
|
||||
portal:
|
||||
enabled: true
|
||||
Loading…
Reference in New Issue