"git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "ad88563bda7066d639e5f63473cf363ff14dde01"
Unverified Commit d703eaae authored by Younes Belkada's avatar Younes Belkada Committed by GitHub
Browse files

[`bnb`] Fix bnb slow tests (#28788)

fix bnb slow tests
parent 74c9cfea
......@@ -342,6 +342,8 @@ class BitsAndBytesConfig(QuantizationConfigMixin):
"""
output = copy.deepcopy(self.__dict__)
output["bnb_4bit_compute_dtype"] = str(output["bnb_4bit_compute_dtype"]).split(".")[1]
output["load_in_4bit"] = self.load_in_4bit
output["load_in_8bit"] = self.load_in_8bit
return output
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment