Image to prompt with BLIP and CLIP
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 

37 lines
695 B

bundled_packages_dir: packages
data_dir: data
description: null
environment_variables: {}
examples_filename: examples.yaml
external_package_dirs:
- ../clip_interrogator
input_type: Any
live_reload: false
model_class_filename: model.py
model_class_name: Model
model_framework: custom
model_metadata: {}
model_module_dir: model
model_name: null
model_type: custom
python_version: py39
requirements:
- torch
- torchvision
- Pillow
- requests
- safetensors
- tqdm
- open_clip_torch
- accelerate
- transformers>=4.27.1
resources:
cpu: 7500m
memory: 15Gi
use_gpu: true
accelerator: A10G
secrets: {}
spec_version: "2.0"
system_packages:
- libgl1-mesa-glx
- libglib2.0-0