How to use skyadmin/cog-webui-sd with Transformers:
# pip install -U transformers accelerate # Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("skyadmin/cog-webui-sd", device_map="auto")
9fd6393
1
2
3
4
5
6
from setuptools import setup if __name__ == '__main__': setup()
