FoodDesert commited on
Commit
08bc111
1 Parent(s): 80f1d22

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ def convert_embedding(sd15_embedding):
8
 
9
  print("got here 1")
10
 
11
- sd15_embedding = torch.load(sd15_embedding.name, weights_only=True)
12
 
13
 
14
  print("got here 2")
 
8
 
9
  print("got here 1")
10
 
11
+ sd15_embedding = torch.load(sd15_embedding.name, map_location=torch.device('cpu'), weights_only=True)
12
 
13
 
14
  print("got here 2")