Spaces:

edwardjiang
/

ecnu-icalk-educhat-sft-002-7b

Runtime error

edwardjiang commited on Jul 22, 2023

Commit

524c5a8

•

1 Parent(s): 89b6055

Update launch.py

Files changed (1) hide show

launch.py CHANGED Viewed

@@ -133,7 +133,7 @@ EduChat的能力
 print('Loading model...')
-model = get_specific_model("models/ecnu-icalk/educhat-sft-002-7b")
 model.half().cuda()
 model.gradient_checkpointing_enable()  # reduce number of stored activations

 print('Loading model...')
+model = get_specific_model(args.model_path)
 model.half().cuda()
 model.gradient_checkpointing_enable()  # reduce number of stored activations