multimodalart HF staff commited on
Commit
4ed3fa9
1 Parent(s): ba35348

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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;}