-
Notifications
You must be signed in to change notification settings - Fork 0
Open
Description
按照readme正确安装环境后进行推理,推理代码似乎进入死循环,无法得到结果,但程序也不报错,显存也占有30多G。请问这个应该如何解决呢?
IP-Prompter/search_images.py:81: UserWarning: The use of x.T on tensors of dimension other than 2 to reverse their shape is deprecated and it will throw an error in a future release. Consider x.mT to transpose batches of matrices or x.permute(*torch.arange(x.ndim - 1, -1, -1)) to reverse the dimensions of a tensor. (Triggered internally at ../aten/src/ATen/native/TensorShape.cpp:3697.)
sims = {file: (feat @ text_features[i].T).item() for file, feat in image_features.items() if file not in used_files}
Metadata
Metadata
Assignees
Labels
No labels