Spaces:
Running
on
Zero
Running
on
Zero
gokaygokay
commited on
Commit
•
8bc7e88
1
Parent(s):
4479682
Update app.py
Browse files
app.py
CHANGED
@@ -14,7 +14,7 @@ original_load = torch.load
|
|
14 |
torch.load = lambda *args, **kwargs: original_load(*args, **kwargs, map_location=torch.device('cpu'))
|
15 |
|
16 |
# Initialize the AuraSR model
|
17 |
-
aura_sr = AuraSR.from_pretrained("fal
|
18 |
|
19 |
# Restore original torch.load
|
20 |
torch.load = original_load
|
|
|
14 |
torch.load = lambda *args, **kwargs: original_load(*args, **kwargs, map_location=torch.device('cpu'))
|
15 |
|
16 |
# Initialize the AuraSR model
|
17 |
+
aura_sr = AuraSR.from_pretrained("fal/AuraSR-v2")
|
18 |
|
19 |
# Restore original torch.load
|
20 |
torch.load = original_load
|