masharpe commited on
Commit
2b37d53
1 Parent(s): fad10f4

Update requirements

Browse files
Files changed (2) hide show
  1. app.py +2 -1
  2. requirements.txt +3 -3
app.py CHANGED
@@ -4,6 +4,7 @@ import spaces
4
  import logging
5
  import time
6
  import html
 
7
  import numpy as np
8
  import gradio as gr
9
  import util
@@ -75,7 +76,7 @@ def format_response_pair(tokens_a, tokens_b):
75
  HEADERS = ['Response (Left)', 'Response (Right)']
76
  repo_id = "Qwen/Qwen2-0.5B-Instruct"
77
 
78
- DRY_RUN = True
79
 
80
  if DRY_RUN:
81
  from load import load_tokenizer
 
4
  import logging
5
  import time
6
  import html
7
+ import os
8
  import numpy as np
9
  import gradio as gr
10
  import util
 
76
  HEADERS = ['Response (Left)', 'Response (Right)']
77
  repo_id = "Qwen/Qwen2-0.5B-Instruct"
78
 
79
+ DRY_RUN = os.environ.get('DRY_RUN') == '1'
80
 
81
  if DRY_RUN:
82
  from load import load_tokenizer
requirements.txt CHANGED
@@ -1,5 +1,5 @@
1
- huggingface_hub==0.22.2
2
  numpy==1.26.4
3
  torch==2.2.0
4
- transformers==4.40.2
5
- accelerate==0.31.0
 
1
+ huggingface_hub==0.24.6
2
  numpy==1.26.4
3
  torch==2.2.0
4
+ transformers==4.44.2
5
+ accelerate==0.33.0