change bnb tests (#34713)
* fix training tests
* fix xpu check
Signed-off-by: jiqing-feng <jiqing.feng@intel.com>
* rm pdb
Signed-off-by: jiqing-feng <jiqing.feng@intel.com>
* fix 4bit logits check
Signed-off-by: jiqing-feng <jiqing.feng@intel.com>
* fix 4bit logits check
Signed-off-by: jiqing-feng <jiqing.feng@intel.com>
* add xpu check on int8 training
* fix training tests
* add llama test on bnb
Signed-off-by: jiqing-feng <jiqing.feng@intel.com>
* only cpu and xpu disable autocast training
Signed-off-by: jiqing-feng <jiqing.feng@intel.com>
* fix format
Signed-off-by: jiqing-feng <jiqing.feng@intel.com>
---------
Signed-off-by: jiqing-feng <jiqing.feng@intel.com>
Co-authored-by: Titus <9048635+Titus-von-Koeller@users.noreply.github.com>
