Commit
·
cdc5960
1
Parent(s):
0cac0a3
Update app.py
Browse files
app.py
CHANGED
@@ -9,7 +9,7 @@ tokenizer = GPT2TokenizerFast.from_pretrained("AlexWortega/instruct_rugptlarge")
|
|
9 |
special_tokens_dict = {'additional_special_tokens': ['<code>', '</code>', '<instructionS>', '<instructionE>', '<next>']}
|
10 |
|
11 |
tokenizer.add_special_tokens(special_tokens_dict)
|
12 |
-
device = '
|
13 |
model = GPT2LMHeadModel.from_pretrained("AlexWortega/instruct_rugptlarge")
|
14 |
#
|
15 |
|
@@ -96,4 +96,4 @@ g = gr.Interface(
|
|
96 |
|
97 |
|
98 |
g.queue(concurrency_count=5)
|
99 |
-
g.launch(
|
|
|
9 |
special_tokens_dict = {'additional_special_tokens': ['<code>', '</code>', '<instructionS>', '<instructionE>', '<next>']}
|
10 |
|
11 |
tokenizer.add_special_tokens(special_tokens_dict)
|
12 |
+
device = 'cpu' # мэх дорога
|
13 |
model = GPT2LMHeadModel.from_pretrained("AlexWortega/instruct_rugptlarge")
|
14 |
#
|
15 |
|
|
|
96 |
|
97 |
|
98 |
g.queue(concurrency_count=5)
|
99 |
+
g.launch()
|