JoJosmin commited on
Commit
199a6fa
ยท
verified ยท
1 Parent(s): b81fb70

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -18,8 +18,8 @@ def load_clip_model():
18
  tokenizer = open_clip.get_tokenizer('hf-hub:Marqo/marqo-fashionSigLIP')
19
 
20
  # ํŒŒ์ธํŠœ๋‹ํ•œ ๋ชจ๋ธ์˜ state_dict ๋ถˆ๋Ÿฌ์˜ค๊ธฐ
21
- state_dict = torch.load('./accessory_clip.pt', map_location=torch.device('cpu'))
22
- model.load_state_dict(state_dict) # ๋ชจ๋ธ์— state_dict ์ ์šฉ
23
 
24
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
25
  model.to(device)
 
18
  tokenizer = open_clip.get_tokenizer('hf-hub:Marqo/marqo-fashionSigLIP')
19
 
20
  # ํŒŒ์ธํŠœ๋‹ํ•œ ๋ชจ๋ธ์˜ state_dict ๋ถˆ๋Ÿฌ์˜ค๊ธฐ
21
+ #state_dict = torch.load('./accessory_clip.pt', map_location=torch.device('cpu'))
22
+ #model.load_state_dict(state_dict) # ๋ชจ๋ธ์— state_dict ์ ์šฉ
23
 
24
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
25
  model.to(device)