Spaces:
Runtime error
Runtime error
Commit
•
ade511f
1
Parent(s):
ad09580
Update app.py
Browse files
app.py
CHANGED
|
@@ -1,5 +1,7 @@
|
|
| 1 |
import gradio as gr
|
| 2 |
from rwkv.model import RWKV
|
|
|
|
|
|
|
| 3 |
from huggingface_hub import hf_hub_download
|
| 4 |
os.environ["RWKV_JIT_ON"] = '1'
|
| 5 |
os.environ["RWKV_CUDA_ON"] = '1'
|
|
|
|
| 1 |
import gradio as gr
|
| 2 |
from rwkv.model import RWKV
|
| 3 |
+
import os, gc, copy, torch, re
|
| 4 |
+
from datetime import datetime
|
| 5 |
from huggingface_hub import hf_hub_download
|
| 6 |
os.environ["RWKV_JIT_ON"] = '1'
|
| 7 |
os.environ["RWKV_CUDA_ON"] = '1'
|