mirror of
https://github.com/onestardao/WFGY.git
synced 2026-04-28 03:29:51 +00:00
13 lines
265 B
Python
13 lines
265 B
Python
# 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)
|