Hugging Face's logo Hugging Face
  • Models
  • Datasets
  • Spaces
  • Buckets new
  • Docs
  • Enterprise
  • Pricing

  • Log In
  • Sign Up

xhyi
/
CodeGen-2B-Multi

Text Generation
Transformers
PyTorch
English
codegen
text generation
causal-lm
custom_code
Model card Files Files and versions
xet
Community
1

Instructions to use xhyi/CodeGen-2B-Multi with libraries, inference providers, notebooks, and local apps. Follow these links to get started.

  • Libraries
  • Transformers

    How to use xhyi/CodeGen-2B-Multi with Transformers:

    # Use a pipeline as a high-level helper
    from transformers import pipeline
    
    pipe = pipeline("text-generation", model="xhyi/CodeGen-2B-Multi", trust_remote_code=True)
    # Load model directly
    from transformers import AutoTokenizer, AutoModelForCausalLM
    
    tokenizer = AutoTokenizer.from_pretrained("xhyi/CodeGen-2B-Multi", trust_remote_code=True)
    model = AutoModelForCausalLM.from_pretrained("xhyi/CodeGen-2B-Multi", trust_remote_code=True)
  • Notebooks
  • Google Colab
  • Kaggle
  • Local Apps
  • vLLM

    How to use xhyi/CodeGen-2B-Multi with vLLM:

    Install from pip and serve model
    # Install vLLM from pip:
    pip install vllm
    # Start the vLLM server:
    vllm serve "xhyi/CodeGen-2B-Multi"
    # Call the server using curl (OpenAI-compatible API):
    curl -X POST "http://localhost:8000/v1/completions" \
    	-H "Content-Type: application/json" \
    	--data '{
    		"model": "xhyi/CodeGen-2B-Multi",
    		"prompt": "Once upon a time,",
    		"max_tokens": 512,
    		"temperature": 0.5
    	}'
    Use Docker
    docker model run hf.co/xhyi/CodeGen-2B-Multi
  • SGLang

    How to use xhyi/CodeGen-2B-Multi 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 "xhyi/CodeGen-2B-Multi" \
        --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": "xhyi/CodeGen-2B-Multi",
    		"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 "xhyi/CodeGen-2B-Multi" \
            --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": "xhyi/CodeGen-2B-Multi",
    		"prompt": "Once upon a time,",
    		"max_tokens": 512,
    		"temperature": 0.5
    	}'
  • Docker Model Runner

    How to use xhyi/CodeGen-2B-Multi with Docker Model Runner:

    docker model run hf.co/xhyi/CodeGen-2B-Multi
CodeGen-2B-Multi
11.3 GB
Ctrl+K
Ctrl+K
  • 1 contributor
History: 7 commits
xhyi's picture
xhyi
Upload modeling_codegen.py
bf4b5c3 almost 4 years ago
  • .gitattributes
    1.17 kB
    initial commit almost 4 years ago
  • README.md
    2.42 kB
    Create README.md almost 4 years ago
  • added_tokens.json
    1.02 kB
    add tokenizer almost 4 years ago
  • config.json
    1.15 kB
    add model almost 4 years ago
  • configuration_codegen.py
    2.73 kB
    Upload configuration_codegen.py almost 4 years ago
  • merges.txt
    456 kB
    add tokenizer almost 4 years ago
  • modeling_codegen.py
    27.3 kB
    Upload modeling_codegen.py almost 4 years ago
  • pytorch_model-00001-of-00002.bin

    Detected Pickle imports (4)

    • "torch.FloatStorage",
    • "torch._utils._rebuild_tensor_v2",
    • "collections.OrderedDict",
    • "torch.BoolStorage"

    What is a pickle import?

    10.1 GB
    xet
    add model almost 4 years ago
  • pytorch_model-00002-of-00002.bin

    Detected Pickle imports (4)

    • "torch.FloatStorage",
    • "torch._utils._rebuild_tensor_v2",
    • "torch.BoolStorage",
    • "collections.OrderedDict"

    What is a pickle import?

    1.16 GB
    xet
    add model almost 4 years ago
  • pytorch_model.bin.index.json
    24.5 kB
    add model almost 4 years ago
  • special_tokens_map.json
    110 Bytes
    add tokenizer almost 4 years ago
  • tokenizer_config.json
    634 Bytes
    add tokenizer almost 4 years ago
  • vocab.json
    899 kB
    add tokenizer almost 4 years ago