Update app.py
Browse files
app.py
CHANGED
|
@@ -12,6 +12,7 @@ from tqdm import tqdm
|
|
| 12 |
# Set environment variables
|
| 13 |
os.environ['RWKV_JIT_ON'] = '1'
|
| 14 |
os.environ["RWKV_CUDA_ON"] = '0'
|
|
|
|
| 15 |
|
| 16 |
# Model options
|
| 17 |
MODELS = {
|
|
|
|
| 12 |
# Set environment variables
|
| 13 |
os.environ['RWKV_JIT_ON'] = '1'
|
| 14 |
os.environ["RWKV_CUDA_ON"] = '0'
|
| 15 |
+
os.environ["RWKV_V7_ON"] = '1'
|
| 16 |
|
| 17 |
# Model options
|
| 18 |
MODELS = {
|