bluestyle97 commited on
Commit
aeea06f
1 Parent(s): 498625f

Update freesplatter/webui/runner.py

Browse files
Files changed (1) hide show
  1. freesplatter/webui/runner.py +2 -1
freesplatter/webui/runner.py CHANGED
@@ -148,7 +148,8 @@ class FreeSplatterRunner:
148
  near=0.01,
149
  far=100,
150
  ssaa=1,
151
- texture_filter='linear-mipmap-linear').to(device)
 
152
 
153
  @torch.inference_mode()
154
  def run_segmentation(
 
148
  near=0.01,
149
  far=100,
150
  ssaa=1,
151
+ texture_filter='linear-mipmap-linear',
152
+ device=device).to(device)
153
 
154
  @torch.inference_mode()
155
  def run_segmentation(