benjolo commited on
Commit
a03c91a
1 Parent(s): b94f032

Update backend/main.py

Browse files
Files changed (1) hide show
  1. 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