Spaces:

gokaygokay
/

AuraSR-v2

Running on Zero

gokaygokay commited on Jun 26

Commit

13446b7

•

1 Parent(s): 39ffbcb

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ import torch
 import spaces
 # Initialize the AuraSR model
-aura_sr = AuraSR.from_pretrained("fal-ai/AuraSR")
 # Move the model to CUDA if available, otherwise keep it on CPU
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

 import spaces
 # Initialize the AuraSR model
+aura_sr = AuraSR.from_pretrained("fal-ai/AuraSR", map_location=torch.device('cpu'))
 # Move the model to CUDA if available, otherwise keep it on CPU
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")