Pierre Chapuis commited on
Commit
8e1b7c8
·
unverified ·
1 Parent(s): b5c8a75

change repo name

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -37,7 +37,7 @@ DTYPE = torch.bfloat16 if torch.cuda.is_bf16_supported() else torch.float32
37
 
38
  # CPU -> GPU dance because of ZeroGPU
39
 
40
- path = "finegrain/weights-light-switcher-space"
41
  model = light_switcher_lite.Model.from_pretrained(path, device=DEVICE_CPU, dtype=DTYPE)
42
 
43
  model.to(DEVICE)
 
37
 
38
  # CPU -> GPU dance because of ZeroGPU
39
 
40
+ path = "finegrain/merged-turbo-light-switcher"
41
  model = light_switcher_lite.Model.from_pretrained(path, device=DEVICE_CPU, dtype=DTYPE)
42
 
43
  model.to(DEVICE)