TruVlad commited on
Commit
d9a8893
·
verified ·
1 Parent(s): dfc14bc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -11
app.py CHANGED
@@ -7,17 +7,17 @@ from flask_apscheduler import APScheduler
7
  from flask import Flask
8
  from flask import request
9
 
10
- model_name = "inclusionAI/Ring-mini-2.0"
11
- print(f"load model {model_name}")
12
- model = AutoModelForCausalLM.from_pretrained(
13
- model_name,
14
- torch_dtype="auto",
15
- device_map="auto",
16
- trust_remote_code=True
17
- )
18
-
19
- print(f"load tokenizer {model_name}")
20
- tokenizer = AutoTokenizer.from_pretrained(model_name)
21
 
22
  app = Flask(__name__)
23
  scheduler = APScheduler()
 
7
  from flask import Flask
8
  from flask import request
9
 
10
+ # model_name = "inclusionAI/Ring-mini-2.0"
11
+ # print(f"load model {model_name}")
12
+ # model = AutoModelForCausalLM.from_pretrained(
13
+ # model_name,
14
+ # torch_dtype="auto",
15
+ # device_map="auto",
16
+ # trust_remote_code=True
17
+ # )
18
+
19
+ # print(f"load tokenizer {model_name}")
20
+ # tokenizer = AutoTokenizer.from_pretrained(model_name)
21
 
22
  app = Flask(__name__)
23
  scheduler = APScheduler()