Update requirements
Browse files- app.py +2 -1
- 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 =
|
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.
|
2 |
numpy==1.26.4
|
3 |
torch==2.2.0
|
4 |
-
transformers==4.
|
5 |
-
accelerate==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
|