Skip to content

Instantly share code, notes, and snippets.

@jerrydotlam

jerrydotlam/CodeLlama-7b-Instruct-hf.md

Forked from chunhualiao/CodeLlama-7b-Instruct-hf.md
Created April 16, 2024 12:17
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save jerrydotlam/b6c58d8a18231e543007d13fe8dd9d91 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save jerrydotlam/b6c58d8a18231e543007d13fe8dd9d91 to your computer and use it in GitHub Desktop.
Download ZIP

Revisions

  1. @chunhualiao chunhualiao revised this gist Sep 3, 2023. 1 changed file with 2 additions and 2 deletions.
    4 changes: 2 additions & 2 deletions CodeLlama-7b-Instruct-hf.md
    View file
    Original file line number Diff line number Diff line change
    @@ -1,7 +1,7 @@
    Based on discussions on https://huggingface.co/codellama/CodeLlama-7b-Instruct-hf/discussions/10

    pip install git+https://github.com/huggingface/transformers.git@main
    pip install tokenizer transformers
    - pip install git+https://github.com/huggingface/transformers.git@main
    - pip install tokenizer transformers


    ```
  2. @chunhualiao chunhualiao created this gist Sep 3, 2023.
    23 changes: 23 additions & 0 deletions CodeLlama-7b-Instruct-hf.md
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,23 @@
    Based on discussions on https://huggingface.co/codellama/CodeLlama-7b-Instruct-hf/discussions/10

    pip install git+https://github.com/huggingface/transformers.git@main
    pip install tokenizer transformers


    ```
    # Use a pipeline as a high-level helper
    from transformers import pipeline
    from transformers import AutoModelForCausalLM, AutoTokenizer, AutoConfig
    tokenizer = AutoTokenizer.from_pretrained("codellama/CodeLlama-7b-Instruct-hf")
    model = AutoModelForCausalLM.from_pretrained("codellama/CodeLlama-7b-Instruct-hf")
    # Create a pipeline
    code_generator = pipeline('text-generation', model=model, tokenizer=tokenizer)
    # Generate code for an input string
    input_string = "Write a python function to calculate the factorial of a number"
    generated_code = code_generator(input_string, max_length=100)[0]['generated_text']
    print(generated_code)
    ```