text-to-image / README.md
osanseviero's picture
Create README.md
3e24795
|
raw
history blame
829 Bytes
metadata
tags:
  - text-to-image
library_name: generic

Text To Image repository template

This is a template repository for text to image to support generic inference with Hugging Face Hub generic Inference API. There are two required steps

  1. Specify the requirements by defining a requirements.txt file.
  2. Implement the pipeline.py __init__ and __call__ methods. These methods are called by the Inference API. The __init__ method should load the model and preload all the elements needed for inference (model, processors, tokenizers, etc.). This is only called once. The __call__ method performs the actual inference. Make sure to follow the same input/output specifications defined in the template for the pipeline to work.

Example repos