I need help downloading 3.1 in colab. I'm not a programmer so I tango with ChatGPT-4o.

I tried !pip install kandinsky3 : error
I tried !pip install git+https://github.com/author/kandinsky3.git

So I did:

!git clone https://github.com/ai-forever/Kandinsky-3.git
%cd Kandinsky-3

and I got : ERROR: Directory '.' is not installable. Neither 'setup.py' nor 'pyproject.toml' found.

In my next cell I have:

import torch
from kandinsky3 import get_pipeline # Import the correct function

Set the device to CUDA

device_map = torch.device('cuda:0')

Define the data types for different components

dtype_map = {
'unet': torch.float32,
'text_encoder': torch.float16,
'movq': torch.float32,
}

Initialize the T2I Flash pipeline for Kandinsky 3.1

t2i_pipe = get_pipeline(device=device_map, dtype=dtype_map)

Move the pipeline to GPU, if available

t2i_pipe = t2i_pipe.to('cuda')

and I get:

ImportError Traceback (most recent call last)
in <cell line: 5>()
3 get_ipython().system('pip install omegaconf # Install omegaconf required by kandinsky3')
4 import torch
----> 5 from kandinsky3.model import get_T2I_Flash_pipeline
6
7 # Set the device to CUDA

ImportError: cannot import name 'get_T2I_Flash_pipeline' from 'kandinsky3.model' (/content/Kandinsky-3/kandinsky3/model/init.py)

I think I can manage the creation of pictures, but I can't find how to import the model...

The examples you provide don't tell how to achieve this.

Please help!! I love Kandinsky and I want to make a colab for everyone...

The colab: https://colab.research.google.com/drive/1bCs9c6HJQS7HD5pQzZmfp4M1wuSQbdbb?usp=sharing

My colab for K 3.0: https://colab.research.google.com/drive/1Tf7rKQ8Yq0zgwsjUbJ11P4Maxzd_M7QL?usp=sharing