Spaces:
Running
Running
Update app.py
Browse files
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("
|
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()}
|