Spaces:
Runtime error
Runtime error
FoodDesert
commited on
Commit
•
08bc111
1
Parent(s):
80f1d22
Update app.py
Browse files
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")
|