mirror of
https://github.com/onestardao/WFGY.git
synced 2026-04-28 03:29:51 +00:00
Create test_sdk_full.py
This commit is contained in:
parent
42cc74024d
commit
db08eea67c
1 changed files with 13 additions and 0 deletions
13
wfgy_sdk/test_sdk_full.py
Normal file
13
wfgy_sdk/test_sdk_full.py
Normal file
|
|
@ -0,0 +1,13 @@
|
|||
# test_sdk_full.py
|
||||
|
||||
import numpy as np
|
||||
from wfgy_sdk import enable
|
||||
|
||||
model = {
|
||||
"I": np.array([1.2, 0.7, 0.5]),
|
||||
"G": np.array([1.0, 0.6, 0.4]),
|
||||
"state": np.array([0.1, 0.2, 0.3]),
|
||||
"attention_logits": np.array([1.2, 0.9, 1.1])
|
||||
}
|
||||
|
||||
model = enable(model)
|
||||
Loading…
Add table
Add a link
Reference in a new issue