Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -20,7 +20,7 @@ model = AutoModelForCausalLM.from_config(config, trust_remote_code=True).float()
|
|
| 20 |
weights_path = hf_hub_download("HuggingFaceTB/nanowhale-100m", "model.safetensors")
|
| 21 |
state_dict = load_file(weights_path)
|
| 22 |
model.load_state_dict(state_dict, strict=True)
|
| 23 |
-
model = model.
|
| 24 |
|
| 25 |
tokenizer = AutoTokenizer.from_pretrained("HuggingFaceTB/nanowhale-100m")
|
| 26 |
|
|
|
|
| 20 |
weights_path = hf_hub_download("HuggingFaceTB/nanowhale-100m", "model.safetensors")
|
| 21 |
state_dict = load_file(weights_path)
|
| 22 |
model.load_state_dict(state_dict, strict=True)
|
| 23 |
+
model = model.eval()
|
| 24 |
|
| 25 |
tokenizer = AutoTokenizer.from_pretrained("HuggingFaceTB/nanowhale-100m")
|
| 26 |
|