Update app.py

app.py

@@ -33,17 +33,17 @@ for key in state_dict:
         rwkv_state_dict[key[5:]] = state_dict[key].half()
     else:
         visual_state_dict[key] = state_dict[key].half()
-print("rwkv state dict has keys: ", len(rwkv_state_dict))
-print("visual state dict has keys: ", len(visual_state_dict))
 # save 
 vision_local_path = output_dir / f"visual.pth"
 rwkv_local_path = output_dir / f"rwkv.pth"
 torch.save(rwkv_state_dict, rwkv_local_path)
 torch.save(visual_state_dict, vision_local_path)
+print("rwkv state dict has keys: ", len(rwkv_state_dict), "saved to ", rwkv_local_path)
+print("visual state dict has keys: ", len(visual_state_dict), "saved to ", vision_local_path)
 ##########################################################################
 vision_tower_name = 'openai/clip-vit-large-patch14-336'
 
-model = RWKV(model=rwkv_local_path, strategy='cuda fp16')
+model = RWKV(model=str(rwkv_local_path), strategy='cuda fp16')
 from rwkv.utils import PIPELINE, PIPELINE_ARGS
 pipeline = PIPELINE(model, "rwkv_vocab_v20230424")