kinda works!!

This commit is contained in:
wassname
2023-06-16 16:52:28 +08:00
parent 09004dd077
commit fe947cb1e0
2 changed files with 1967 additions and 1 deletions
+1 -1
View File
@@ -58,7 +58,7 @@ if __name__=="__main__":
parser.add_argument('model_repo', type=str)
parser.add_argument('-l', '--lora_repo', type=str, default=None, help='Name of the lora repo')
parser.add_argument('-f', '--force_download', type=str, default=None, help='Name of the lora repo')
parser.add_argument('-r', '--resume_download', type=str, default=None, help='Name of the lora repo')
parser.add_argument('-r', '--resume_download', action='store_true', help='Name of the lora repo')
args = parser.parse_args()
main(args.model_repo, args.lora_repo, force_download=args.force_download, resume_download=args.resume_download, low_cpu_mem_usage=True)