Skip to content
GitLab
Projects Groups Snippets
  • /
  • Help
    • Help
    • Support
    • Community forum
    • Submit feedback
    • Contribute to GitLab
  • Sign in / Register
  • M metaseq
  • Project information
    • Project information
    • Activity
    • Labels
    • Members
  • Repository
    • Repository
    • Files
    • Commits
    • Branches
    • Tags
    • Contributors
    • Graph
    • Compare
  • Issues 95
    • Issues 95
    • List
    • Boards
    • Service Desk
    • Milestones
  • Merge requests 41
    • Merge requests 41
  • CI/CD
    • CI/CD
    • Pipelines
    • Jobs
    • Schedules
  • Deployments
    • Deployments
    • Environments
    • Releases
  • Packages and registries
    • Packages and registries
    • Package Registry
    • Infrastructure Registry
  • Monitor
    • Monitor
    • Incidents
  • Analytics
    • Analytics
    • Value stream
    • CI/CD
    • Repository
  • Wiki
    • Wiki
  • Snippets
    • Snippets
  • Activity
  • Graph
  • Create a new issue
  • Jobs
  • Commits
  • Issue Boards
Collapse sidebar
  • Administrator
  • metaseq
  • Merge requests
  • !406

Bugfix - Generation API fails if the generated sequence is truncated.

  • Review changes

  • Download
  • Email patches
  • Plain diff
Merged Administrator requested to merge punitkoura/bug_fix_distribution into main Oct 13, 2022
  • Overview 11
  • Commits 5
  • Pipelines 0
  • Changes 8

Created by: punitkoura

Patch Description

  1. Ensure that we pad the mask properly when calculating the token distribution.
  2. Add a function to decode tokens in utils.py - This will be used internally.

Testing steps Ran an API locally to verify generation works.

curl -k <URL> -H "Authorization: Bearer Punit" -H "Content-Type: application/json" \
> -d '{
> "prompt": [2, 44019, 28049, 13770, 6439, 22824, 2306, 14380, 6804, 5457, 21523, 50118, 50118, 713, 16, 10, 569, 9, 10, 11954, 6439, 3704, 81],
> "temperature": 0.0,
> "max_tokens": 300, "min_tokens": 0,
> "top_p": 1.0, "n": 1, "best_of": 1,
> "echo": true, "logprobs": 1, "seed": 1
> }'
Assignee
Assign to
Reviewers
Request review from
Time tracking
Source branch: punitkoura/bug_fix_distribution