Update README.md
Browse files
README.md
CHANGED
|
@@ -31,6 +31,8 @@ model = AutoModel.from_config(config, trust_remote_code=True)
|
|
| 31 |
# or directly from_pretrained
|
| 32 |
model = AutoModel.from_pretrained(model_path, trust_remote_code=True, device_map="auto")
|
| 33 |
|
|
|
|
|
|
|
| 34 |
files = [
|
| 35 |
f"output/sana_test_prompt/0.png",
|
| 36 |
f"output/sana_test_prompt/1.png"
|
|
|
|
| 31 |
# or directly from_pretrained
|
| 32 |
model = AutoModel.from_pretrained(model_path, trust_remote_code=True, device_map="auto")
|
| 33 |
|
| 34 |
+
yes_id = model.tokenizer.encode("yes", add_special_tokens=False)[0]
|
| 35 |
+
no_id = model.tokenizer.encode("no", add_special_tokens=False)[0]
|
| 36 |
files = [
|
| 37 |
f"output/sana_test_prompt/0.png",
|
| 38 |
f"output/sana_test_prompt/1.png"
|