Spaces:
Paused
Paused
Update backend/main.py
Browse files- backend/main.py +1 -1
backend/main.py
CHANGED
@@ -142,7 +142,7 @@ static_files = {
|
|
142 |
},
|
143 |
}
|
144 |
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
145 |
-
processor = AutoProcessor.from_pretrained("facebook/seamless-m4t-v2-large")
|
146 |
# PM - hardcoding temporarily as my GPU doesnt have enough vram
|
147 |
model = SeamlessM4Tv2Model.from_pretrained("facebook/seamless-m4t-v2-large").to("cpu")
|
148 |
|
|
|
142 |
},
|
143 |
}
|
144 |
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
145 |
+
processor = AutoProcessor.from_pretrained("facebook/seamless-m4t-v2-large", cache_dir="/.cache")
|
146 |
# PM - hardcoding temporarily as my GPU doesnt have enough vram
|
147 |
model = SeamlessM4Tv2Model.from_pretrained("facebook/seamless-m4t-v2-large").to("cpu")
|
148 |
|