/okral-image-model

An image model built on okrol model

Primary LanguagePythonOtherNOASSERTION

Okrol-Image-Model

An image model created on the base of okrol model by me
diffusion_final_image_id_example1 json_0_Sat May 25 14:21:57 2024 diffusion_final_image_id_example1 json_1_Sat May 25 14:35:44 2024


Installation

Install the requirements pip install -r requirements.txt, Install the language processer python -m spacy download en_core_web_sm


Usage

The model is not really great and shouldn't be used for anything else than experimenting:

  1. It losses some data from epoches
  2. It is unstable
  3. It bugs out very often
  4. The diffusion algorithm is not very good

How to add data

This is a blueprint for example data

[
    {
        "text": "A cat sitting on a couch.",
        "image": "images/cat1.jpg"
    }
]

#Note in this example images folder will be in the root folder not the traing_data folder Ai was used for generating docstrings/comments