[Misc] Merge bitsandbytes_stacked_params_mapping and packed_modules_mapping (#11924)
Signed-off-by: Jee Jee Li <pandaleefree@gmail.com>
This commit is contained in:
@@ -1028,13 +1028,6 @@ class QWenLLM(QWenBaseModel):
|
||||
embedding_modules = {}
|
||||
embedding_padding_modules = []
|
||||
|
||||
# BitandBytes specific attributes
|
||||
bitsandbytes_stacked_params_mapping = {
|
||||
# shard_name, weight_name, index
|
||||
"w2": ("gate_up_proj", 0),
|
||||
"w1": ("gate_up_proj", 1),
|
||||
}
|
||||
|
||||
|
||||
class QWenVL(QWenBaseModel, SupportsMultiModal):
|
||||
packed_modules_mapping = {
|
||||
|
||||
Reference in New Issue
Block a user