codeShare commited on
Commit
3212c7e
1 Parent(s): 44eca84

Update sd_token_similarity_calculator.ipynb

Browse files
sd_token_similarity_calculator.ipynb CHANGED
@@ -22,6 +22,7 @@
22
  "import torch\n",
23
  "from torch import linalg as LA\n",
24
  "device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')\n",
 
25
  "token = torch.load('sd15_tensors.pt', map_location=device, weights_only=True)"
26
  ],
27
  "metadata": {
 
22
  "import torch\n",
23
  "from torch import linalg as LA\n",
24
  "device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')\n",
25
+ "%cd content/sd_tokens\n",
26
  "token = torch.load('sd15_tensors.pt', map_location=device, weights_only=True)"
27
  ],
28
  "metadata": {