containers / podman-desktop-extension-ai-lab

Work with LLMs on a local environment using containers
https://podman-desktop.io/extensions/ai-lab
Apache License 2.0
182 stars 40 forks source link

[UX] Create an icon to distinguish fine-tuned models from others. #1569

Open MariaLeonova opened 3 months ago

MariaLeonova commented 3 months ago

Is your enhancement related to a problem? Please describe

Fine-tuned models in the Model catalog will need to be distinguished from the ones that have not been fine-tuned by way of showing an icon. This icon needs to be designed and added to the Podman Desktop AI Lab icon repo.

Describe the solution you'd like

I would like to propose an icon to be used in the catalog.

Describe alternatives you've considered

No response

Additional context

No response

MariaLeonova commented 3 months ago

Related to #1300

MariaLeonova commented 3 months ago

@jeffmaury @ekidneyrh will the action "Tune with InstructLab" in the left menu panel also need an icon? g115

axel7083 commented 2 months ago

Finding an icon for the fine tuned models is not an easy task, as they are not much example in the wild due to how new it is.

Fine tuning

I would argue, first let's look at the definition of fine tuning: _fine-tuning is an approach to transfer learning in which the parameters of a pre-trained model are trained on new data._^1

We fine tune an existing model to add on top of the existing specification, to have it suit our needs.

For example, I can fine tuned a model for the classification of documents, and use my company history of user classification to give some background to the model and improve its result on reproducing the expecting task(s).

Existing icons

image

Google gave us some idea of how currently people are representing it, with a 🪛 .

We could imagine trying to combine this concept of Screwdriver with the current icon.

MariaLeonova commented 2 months ago

So before we bring any extra elements in, I would suggest trying out options having to do with size, colour and shape. The idea is that when you are fine-tuning you are changing the original model a bit, to suit your needs - adding knowledge to it (or your personal docs, whichever). So in the same fashion I thought we could change the model icon - all of it - or parts - to be different shape / colour / size circles: Screenshot 2024-09-10 at 14 40 34

Close-up on icons: Screenshot 2024-09-10 at 14 41 45

ekidneyrh commented 2 months ago

I quite like the idea of playing with the shape like this one:

image

it makes me feel like I'm molding the model physically which it a good visual for this! I think we could play around with this more - maybe even curving the lines? Here's a v quick messy idea I made:

image

MariaLeonova commented 1 month ago

Continuing with the topic of physically changing the model, here's a few of the less wild options for the icon:

Screenshot 2024-09-27 at 16 10 50

@ekidneyrh @jeffmaury @axel7083

jeffmaury commented 1 month ago

Continuing with the topic of physically changing the model, here's a few of the less wild options for the icon:

Screenshot 2024-09-27 at 16 10 50

@ekidneyrh @jeffmaury @axel7083

Either the left one or the original one with curved lines for my POV

ekidneyrh commented 1 month ago

I like the first and second one! I already said why i like the first above.

For the second I like how symmetrical it is - it makes it feel polished ('fine tuned') imo.

MariaLeonova commented 1 month ago

As I personally prefer the one on the left, too, let's go with it (unless someone objects)?

@jeffmaury what format and size file would be needed for implementation, please?

jeffmaury commented 1 month ago

As I personally prefer the one on the left, too, let's go with it (unless someone objects)?

@jeffmaury what format and size file would be needed for implementation, please?

SVG 40x40

MariaLeonova commented 1 month ago

@jeffmaury let me know if these are ok and if you'd like them uploaded to github somewhere!

Light mode: fine-tuned-icon-40x40-light

Dark mode: fine-tuned-icon-40x40-dark