fix caption fl

2023-07-03 15:52:50 -04:00 · 2023-07-03 15:52:50 -04:00 · 6907c01b51
parent 2afea3bdb8
commit 6907c01b51
1 changed files with 1 additions and 1 deletions
--- a/caption_fl.py
+++ b/caption_fl.py
@ -82,7 +82,7 @@ def get_dtype_for_cuda_device(device):

 def main(args):
    device = "cuda" if torch.cuda.is_available() and not args.force_cpu else "cpu"
-    dtype = get_dtype_for_cuda_device() if device == "cuda" else torch.float32
+    dtype = get_dtype_for_cuda_device(device) if device == "cuda" else torch.float32

    if args.prompt:
        prompt = args.prompt