Instructions to use akhooli/gpt2-small-arabic with libraries, inference providers, notebooks, and local apps. Follow these links to get started.

Libraries

How to use akhooli/gpt2-small-arabic with Transformers:

# Use a pipeline as a high-level helper
from transformers import pipeline

pipe = pipeline("text-generation", model="akhooli/gpt2-small-arabic")

# Load model directly
from transformers import AutoTokenizer, AutoModelForCausalLM

tokenizer = AutoTokenizer.from_pretrained("akhooli/gpt2-small-arabic")
model = AutoModelForCausalLM.from_pretrained("akhooli/gpt2-small-arabic")

Notebooks
Google Colab
Kaggle
Local Apps

vLLM

How to use akhooli/gpt2-small-arabic with vLLM:

Install from pip and serve model

# Install vLLM from pip:
pip install vllm
# Start the vLLM server:
vllm serve "akhooli/gpt2-small-arabic"
# Call the server using curl (OpenAI-compatible API):
curl -X POST "http://localhost:8000/v1/completions" \
	-H "Content-Type: application/json" \
	--data '{
		"model": "akhooli/gpt2-small-arabic",
		"prompt": "Once upon a time,",
		"max_tokens": 512,
		"temperature": 0.5
	}'

Use Docker

docker model run hf.co/akhooli/gpt2-small-arabic

SGLang

How to use akhooli/gpt2-small-arabic with SGLang:

Install from pip and serve model

# Install SGLang from pip:
pip install sglang
# Start the SGLang server:
python3 -m sglang.launch_server \
    --model-path "akhooli/gpt2-small-arabic" \
    --host 0.0.0.0 \
    --port 30000
# Call the server using curl (OpenAI-compatible API):
curl -X POST "http://localhost:30000/v1/completions" \
	-H "Content-Type: application/json" \
	--data '{
		"model": "akhooli/gpt2-small-arabic",
		"prompt": "Once upon a time,",
		"max_tokens": 512,
		"temperature": 0.5
	}'

Use Docker images

docker run --gpus all \
    --shm-size 32g \
    -p 30000:30000 \
    -v ~/.cache/huggingface:/root/.cache/huggingface \
    --env "HF_TOKEN=<secret>" \
    --ipc=host \
    lmsysorg/sglang:latest \
    python3 -m sglang.launch_server \
        --model-path "akhooli/gpt2-small-arabic" \
        --host 0.0.0.0 \
        --port 30000
# Call the server using curl (OpenAI-compatible API):
curl -X POST "http://localhost:30000/v1/completions" \
	-H "Content-Type: application/json" \
	--data '{
		"model": "akhooli/gpt2-small-arabic",
		"prompt": "Once upon a time,",
		"max_tokens": 512,
		"temperature": 0.5
	}'

Docker Model Runner
How to use akhooli/gpt2-small-arabic with Docker Model Runner:
```
docker model run hf.co/akhooli/gpt2-small-arabic
```

YAML Metadata Error:"datasets[0]" with value "Arabic Wikipedia" is not valid. If possible, use a dataset id from https://hf.co/datasets.

GPT2-Small-Arabic

Model description

GPT2 model from Arabic Wikipedia dataset based on gpt2-small (using Fastai2).

Intended uses & limitations

How to use

An example is provided in this colab notebook. Both text and poetry (fine-tuned model) generation are included.

Limitations and bias

GPT2-small-arabic (trained on Arabic Wikipedia) has several limitations in terms of coverage (Arabic Wikipeedia quality, no diacritics) and training performance. Use as demonstration or proof of concepts but not as production code.

Training data

This pretrained model used the Arabic Wikipedia dump (around 900 MB).

Training procedure

Training was done using Fastai2 library on Kaggle, using free GPU.

Eval results

Final perplexity reached was 72.19, loss: 4.28, accuracy: 0.307

BibTeX entry and citation info

@inproceedings{Abed Khooli,
  year={2020}
}

Downloads last month: 2,672

Safetensors

Model size

0.1B params

Tensor type

F32

Model tree for akhooli/gpt2-small-arabic

Adapters

1 model

Finetunes