Code

PTQ memory optimization#11257

Code

Merged

janekl merged 15 commits intomainNVIDIA/NeMo:mainfrom pikaminski/nemo2_ptq_memory_optimizationNVIDIA/NeMo:pikaminski/nemo2_ptq_memory_optimizationCopy head branch name to clipboardNov 18, 2024

+41-20

Commits

Commits on Nov 12, 2024

• Initial commit

  Show description for ed092ee
  Laplasjan107

  committed
  ed092ee

  Copy full SHA for ed092ee

• Apply isort and black reformatting

  Show description for 621d9f0
  Laplasjan107

  committed
  621d9f0

  Copy full SHA for 621d9f0

• Add sample generate

  Show description for fae0f8c
  Laplasjan107

  committed
  fae0f8c

  Copy full SHA for fae0f8c

• Apply isort and black reformatting

  Show description for ccf15c7
  Laplasjan107

  committed
  ccf15c7

  Copy full SHA for ccf15c7

Commits on Nov 13, 2024

• Nemotron quantization, reduce diff

  Show description for 8296e37
  Laplasjan107

  committed
  8296e37

  Copy full SHA for 8296e37

• Apply isort and black reformatting

  Show description for 8a63a72
  Laplasjan107

  committed
  8a63a72

  Copy full SHA for 8a63a72

• Merge branch 'main' into pikaminski/nemo2_ptq_memory_optimization

  Show description for 94c0081
  Laplasjan107

  authored
  94c0081

  Copy full SHA for 94c0081

• Apply isort and black reformatting

  Show description for a05470a
  Laplasjan107

  committed
  a05470a

  Copy full SHA for a05470a

• Reduce diff

  Show description for 0e7a8dd
  Laplasjan107

  committed
  0e7a8dd

  Copy full SHA for 0e7a8dd

Commits on Nov 14, 2024

• code review suggestions

  Show description for 3ce6608
  Laplasjan107

  committed
  3ce6608

  Copy full SHA for 3ce6608

• Bug fixes

  Show description for 1a58606
  Laplasjan107

  committed
  1a58606

  Copy full SHA for 1a58606

• remove not needed import

  Show description for dd6f611
  Laplasjan107

  committed
  dd6f611

  Copy full SHA for dd6f611

Commits on Nov 18, 2024

• fix model type and allow ddp/optim setup

  Show description for 77876cb
  Laplasjan107

  committed
  77876cb

  Copy full SHA for 77876cb

• Apply isort and black reformatting

  Show description for ba4925a
  Laplasjan107

  committed
  ba4925a

  Copy full SHA for ba4925a

• Merge branch 'main' into pikaminski/nemo2_ptq_memory_optimization

  janekl

  authored
  587bdae

  Copy full SHA for 587bdae