Spaces:
Sleeping
Sleeping
Commit
•
4ed3fa9
1
Parent(s):
ba35348
Update app.py
Browse files
app.py
CHANGED
@@ -38,7 +38,7 @@ for item in sdxl_loras:
|
|
38 |
state_dict = torch.load(saved_name)
|
39 |
|
40 |
item["saved_name"] = saved_name
|
41 |
-
item["state_dict"] = {k: v.to(device="cuda", dtype=torch.float16) for k, v in state_dict.items() if torch.is_tensor(v)}
|
42 |
|
43 |
css = '''
|
44 |
#title{text-align:center;}
|
|
|
38 |
state_dict = torch.load(saved_name)
|
39 |
|
40 |
item["saved_name"] = saved_name
|
41 |
+
item["state_dict"] = state_dict #{k: v.to(device="cuda", dtype=torch.float16) for k, v in state_dict.items() if torch.is_tensor(v)}
|
42 |
|
43 |
css = '''
|
44 |
#title{text-align:center;}
|