Spaces:
Runtime error
Runtime error
Pierre Chapuis
commited on
change repo name
Browse files
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/
|
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)
|