Edit model card

Model Card for Model ID

Model Details

Model Description

This Model is a first test to combine Jamba architecture with 1.58 bits linear layers and mixture of attention head.

The goal is to developpe and test if this kind of architectures have not too much quality loss for a fast inference.

  • Model type: Mixture of attention head and mixture of expert 1.58bit linear layers
  • License: Apache licence 2.0

Model Sources [optional]

How to Get Started with the Model

If you want to test this model please look at this repo at this commit

Training Details

Training Data

We use the first 100k data of Locutusque/UltraTextbooks to train this model

Training Procedure

We use adam-8 bits with default betas and epsilon values

Preprocessing [optional]

The data fit the model max length i.e. 512 tokens

Training Hyperparameters

Please look at this file to see the hyperparameters

Technical Specifications [optional]

Compute Infrastructure

Hardware

  • one 4070 ti GPU

Software

  • pytorch, transformers etc
Downloads last month
24
Safetensors
Model size
1.06B params
Tensor type
F32
·
Inference Examples
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.

Dataset used to train Ostixe360/Moah-MoE-1.58b-1B