Hugging Face's logo

Spaces:
TamisAI
/
inference-api-g1
Running on CPU Upgrade

App Files Community
alexfremont commited on
Commit
e7fd9ba
·
1 Parent(s): 41517aa

change models

Browse files
Files changed (1) hide show
  1. main.py +1 -6
main.py CHANGED
@@ -41,18 +41,13 @@ models_locations = [
41
  {
42
  "repo_id": "TamisAI/category-lamp",
43
  "subfolder": "maison-charles/corail-152-0001-32-256-old",
44
- "filename": "maison-charles-corail.pth",
45
  },
46
  {
47
  "repo_id": "TamisAI/category-lamp",
48
  "subfolder": "maison-charles/corail-152-0001-32-256",
49
  "filename": "maison-charles-corail.pth",
50
  },
51
- {
52
- "repo_id": "TamisAI/category-lamp",
53
- "subfolder": "maison-charles/corail-50-0001-32-256",
54
- "filename": "maison-charles-corail.pth",
55
- },
56
  ]
57
 
58
  device = torch.device("cpu")
 
41
  {
42
  "repo_id": "TamisAI/category-lamp",
43
  "subfolder": "maison-charles/corail-152-0001-32-256-old",
44
+ "filename": "maison-charles-corail-old.pth",
45
  },
46
  {
47
  "repo_id": "TamisAI/category-lamp",
48
  "subfolder": "maison-charles/corail-152-0001-32-256",
49
  "filename": "maison-charles-corail.pth",
50
  },
 
 
 
 
 
51
  ]
52
 
53
  device = torch.device("cpu")