wusize commited on
Commit
1989442
·
verified ·
1 Parent(s): 792868b

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. .idea/workspace.xml +1 -1
  2. app.py +2 -2
.idea/workspace.xml CHANGED
@@ -54,7 +54,7 @@
54
  <option name="number" value="Default" />
55
  <option name="presentableId" value="Default" />
56
  <updated>1760056680813</updated>
57
- <workItem from="1760056681869" duration="4202000" />
58
  </task>
59
  <servers />
60
  </component>
 
54
  <option name="number" value="Default" />
55
  <option name="presentableId" value="Default" />
56
  <updated>1760056680813</updated>
57
+ <workItem from="1760056681869" duration="4675000" />
58
  </task>
59
  <servers />
60
  </component>
app.py CHANGED
@@ -61,9 +61,9 @@ os.remove('checkpoints/vae.pth')
61
 
62
 
63
  if torch.cuda.is_available():
64
- harmon_model = model.to(torch.bfloat16).cuda()
65
  else:
66
- harmon_model = model.to(torch.float32)
67
 
68
 
69
  def fig_to_image(fig):
 
61
 
62
 
63
  if torch.cuda.is_available():
64
+ model = model.to(torch.bfloat16).cuda()
65
  else:
66
+ model = model.to(torch.float32)
67
 
68
 
69
  def fig_to_image(fig):