Daemontatox commited on
Commit
7e59c62
·
verified ·
1 Parent(s): d5e703b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,4 +1,4 @@
1
- import spaces
2
  import os, copy
3
  os.environ["RWKV_V7_ON"] = '1'
4
  os.environ["RWKV_JIT_ON"] = '1'
@@ -33,7 +33,7 @@ pipeline_v6 = PIPELINE(model_v6, "rwkv_vocab_v20230424")
33
  args = model_v6.args
34
 
35
  penalty_decay = 0.996
36
- @spaces.GPU()
37
  def generate_prompt(instruction, input=""):
38
  instruction = instruction.strip().replace('\r\n','\n').replace('\n\n','\n')
39
  input = input.strip().replace('\r\n','\n').replace('\n\n','\n')
 
1
+
2
  import os, copy
3
  os.environ["RWKV_V7_ON"] = '1'
4
  os.environ["RWKV_JIT_ON"] = '1'
 
33
  args = model_v6.args
34
 
35
  penalty_decay = 0.996
36
+
37
  def generate_prompt(instruction, input=""):
38
  instruction = instruction.strip().replace('\r\n','\n').replace('\n\n','\n')
39
  input = input.strip().replace('\r\n','\n').replace('\n\n','\n')