Spaces:
Running
on
Zero
Running
on
Zero
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;}
|