stanrom
/

internlm-xcomposer-7b-4bit

stanrom commited on May 24

Commit

f77165e

•

1 Parent(s): 7600482

Update modeling_InternLM_XComposer.py

Files changed (1) hide show

modeling_InternLM_XComposer.py CHANGED Viewed

@@ -108,6 +108,7 @@ conversation
         self.eoh = '<TOKENS_UNUSED_0>'  # end of human
         self.eoa = '<TOKENS_UNUSED_1>'  # end of assistant
         stop_words_ids = [
             torch.tensor([103027]).to(config.device),
             torch.tensor([103028]).to(config.device),

         self.eoh = '<TOKENS_UNUSED_0>'  # end of human
         self.eoa = '<TOKENS_UNUSED_1>'  # end of assistant
+        print('config.device =', config.device)
         stop_words_ids = [
             torch.tensor([103027]).to(config.device),
             torch.tensor([103028]).to(config.device),