TruVlad commited on
Commit
87bb67d
·
verified ·
1 Parent(s): 2e124ae
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -1,19 +1,21 @@
1
  from transformers import AutoModelForCausalLM, AutoTokenizer
2
 
3
  model_name = "inclusionAI/Ring-mini-2.0"
4
-
5
  model = AutoModelForCausalLM.from_pretrained(
6
  model_name,
7
  torch_dtype="auto",
8
  device_map="auto",
9
  trust_remote_code=True
10
  )
 
 
11
  tokenizer = AutoTokenizer.from_pretrained(model_name)
12
 
13
  from flask import Flask
14
 
15
  app = Flask(__name__)
16
-
17
  # Главная страница
18
  @app.route("/")
19
  def home():
 
1
  from transformers import AutoModelForCausalLM, AutoTokenizer
2
 
3
  model_name = "inclusionAI/Ring-mini-2.0"
4
+ print(f"load model {model_name}")
5
  model = AutoModelForCausalLM.from_pretrained(
6
  model_name,
7
  torch_dtype="auto",
8
  device_map="auto",
9
  trust_remote_code=True
10
  )
11
+
12
+ print(f"load tokenizer {model_name}")
13
  tokenizer = AutoTokenizer.from_pretrained(model_name)
14
 
15
  from flask import Flask
16
 
17
  app = Flask(__name__)
18
+ print(f"Flask app")
19
  # Главная страница
20
  @app.route("/")
21
  def home():