move model to https://github.com/ssbuild/aigc_zoo

Signed-off-by: ssbuild <[email protected]>

Author: ssbuild

args.MD

@@ -5,8 +5,7 @@
     enable_deepspeed = False
     enable_ptv2 = False
     enable_lora = True
-    enable_int8 = False # qlora int8
-    enable_int4 = False # qlora int4
+    load_in_bit = 0  # 4 load_in_4bit, 8 load_in_8bit  other  0
 
 ### 单机多卡
 ```text

config/reward_config/main.py

@@ -11,28 +11,24 @@
 enable_deepspeed = False
 enable_ptv2 = False
 enable_lora = True
-enable_int8 = False # qlora int8
-enable_int4 = False # qlora int4
+load_in_bit = 0  # 4 load_in_4bit, 8 load_in_8bit  other  0
 
 
 if enable_lora:
     from config.reward_config.reward_config_lora import *
+elif enable_ptv2:
+    raise NotImplemented
 else:
     from config.reward_config.reward_config import *
 
 
 
 if enable_lora:
     enable_ptv2 = False
-    if enable_int4:
-        global_args['load_in_4bit'] = True
-        global_args['load_in_8bit'] = False
-
-    if enable_int8:
-        global_args['load_in_4bit'] = False
-        global_args['load_in_8bit'] = True
+    global_args['load_in_4bit'] = load_in_bit == 4
+    global_args['load_in_8bit'] = load_in_bit == 8
 
-    if not enable_int4:
+    if global_args['load_in_4bit']:
         global_args['quantization_config'] = None
 
     #检查lora adalora是否开启