prompt operation version -- li

This commit is contained in:
memoryunreal
2023-04-26 09:21:44 +00:00
parent 040b6891d6
commit 35850bdb73

4
app.py
View File

@@ -364,8 +364,8 @@ folder ="./checkpoints"
SAM_checkpoint = download_checkpoint(sam_checkpoint_url, folder, sam_checkpoint)
xmem_checkpoint = download_checkpoint(xmem_checkpoint_url, folder, xmem_checkpoint)
e2fgvi_checkpoint = download_checkpoint_from_google_drive(e2fgvi_checkpoint_id, folder, e2fgvi_checkpoint)
args.port = 12212
args.device = "cuda:2"
# args.port = 12212
# args.device = "cuda:2"
# args.mask_save = True
# initialize sam, xmem, e2fgvi models