stanrom
/

ShareCaptioner

stanrom commited on May 19

Commit

7bbe75c

•

1 Parent(s): 5c9b572

load_in_4bit=True

Files changed (1) hide show

configuration_InternLM_XComposer.py CHANGED Viewed

@@ -35,6 +35,7 @@ class InternLMXComposerConfig(PretrainedConfig):
         kqvo_bias=True,
         device='cuda',
         internlm_lora=None,
         **kwargs,
     ):
         self.vocab_size = vocab_size

         kqvo_bias=True,
         device='cuda',
         internlm_lora=None,
+        load_in_4bit=True,
         **kwargs,
     ):
         self.vocab_size = vocab_size