Sakalti commited on
Commit
396f847
Β·
verified Β·
1 Parent(s): 329047b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -34,7 +34,7 @@ def convert_and_deploy(url, repo_id, hf_token):
34
  state_dict = {k: f.get_tensor(k) for k in f.keys()}
35
 
36
  # γƒ’γƒ‡γƒ«γ‚’εˆζœŸεŒ–
37
- model = AutoModel.from_config(AutoModel.from_pretrained("path_to_model", torch_dtype=torch.float16, token=hf_token).config)
38
 
39
  # BF16からFP16に倉換
40
  state_dict = {k: v.to(torch.float16) for k, v in state_dict.items()}
 
34
  state_dict = {k: f.get_tensor(k) for k in f.keys()}
35
 
36
  # γƒ’γƒ‡γƒ«γ‚’εˆζœŸεŒ–
37
+ model = AutoModel.from_config(AutoModel.from_pretrained("Sakalti/Qwen-float16-0.5b", torch_dtype=torch.float16, token=hf_token).config)
38
 
39
  # BF16からFP16に倉換
40
  state_dict = {k: v.to(torch.float16) for k, v in state_dict.items()}