add eval() and update output
Browse files
README.md
CHANGED
@@ -49,6 +49,7 @@ model = AutoModelForCausalLM.from_pretrained(
|
|
49 |
trust_remote_code=True,
|
50 |
)
|
51 |
model.half()
|
|
|
52 |
|
53 |
if torch.cuda.is_available():
|
54 |
model = model.to("cuda")
|
@@ -79,8 +80,8 @@ tokens = model.generate(
|
|
79 |
out = tokenizer.decode(tokens[0], skip_special_tokens=True)
|
80 |
print(out)
|
81 |
"""
|
82 |
-
|
83 |
-
20
|
84 |
"""
|
85 |
```
|
86 |
|
|
|
49 |
trust_remote_code=True,
|
50 |
)
|
51 |
model.half()
|
52 |
+
model.eval()
|
53 |
|
54 |
if torch.cuda.is_available():
|
55 |
model = model.to("cuda")
|
|
|
80 |
out = tokenizer.decode(tokens[0], skip_special_tokens=True)
|
81 |
print(out)
|
82 |
"""
|
83 |
+
AI ใง็งๅญฆ็ ็ฉถใๅ ้ใใใซใฏใใใผใฟ้งๅๅๆๅใๅฟ
่ฆใงใใใใจใๆใใใซใชใฃใฆใใฆใใพใใ็ ็ฉถใฎใใใใๅด้ขใงใใใผใฟใใใ้่ฆใซใชใฃใฆใใใฎใงใใ
|
84 |
+
20 ไธ็ดใฎ็งๅญฆใฏใ็ ็ฉถ่
ใ็ดๆฅ็ ็ฉถใ่กใใใจใงใ็ ็ฉถใใผใฟใๆดป็จใใฆใใพใใใใใฎๅพใๅคใใฎ็งๅญฆๅ้ใงใฏใใผใฟใฏๆๅใงๅๆใใใใใใซใชใฃใใใฎใฎใใใใใฎๆนๆณใซใฏๅคๅคงใชใณในใใจๅดๅใใใใใใจใๅใใใพใใใ ใใใงใๅคใใฎ็ ็ฉถ่
ใ็ ็ฉถ่
ใฐใซใผใใฏใใใๅน็็ใชๆๆณใ้็บใใ็ ็ฉถใฎ่ฆๆจกใๆกๅคงใใฆใใพใใใ21 ไธ็ดใซใชใใจใ็ ็ฉถ่
ใๆๅใงๅฎๆฝใใๅฟ
่ฆใฎใใ็ ็ฉถใฏใใใฎๅคง้จๅใ็ ็ฉถ่
ใ่ชๅๅใงใใใใใซใชใใพใใใ
|
85 |
"""
|
86 |
```
|
87 |
|