SFM2001 commited on
Commit
9925e66
·
1 Parent(s): 626f7ac

try to debug qwen?

Browse files
Files changed (2) hide show
  1. create_app.py +4 -2
  2. templates/base.html +1 -1
create_app.py CHANGED
@@ -27,11 +27,13 @@ def load_models():
27
  LONGFORMER_MODEL = CustomLongformerForSequenceClassification(config).from_pretrained('SFM2001/LongFormerScorer')
28
  LONGFORMER_MODEL = LONGFORMER_MODEL.to(device)
29
  LONGFORMER_MODEL.eval()
30
-
31
  model_name = 'Qwen/Qwen3-1.7B'
32
  QWEN_TOKENIZER = AutoTokenizer.from_pretrained(model_name, device='auto')
33
  QWEN_TOKENIZER.pad_token_id = QWEN_TOKENIZER.eos_token_id
34
- QWEN_MODEL = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float16).half()
 
 
35
  MODELS_LOADED = True
36
  print("LOAD ENDED")
37
 
 
27
  LONGFORMER_MODEL = CustomLongformerForSequenceClassification(config).from_pretrained('SFM2001/LongFormerScorer')
28
  LONGFORMER_MODEL = LONGFORMER_MODEL.to(device)
29
  LONGFORMER_MODEL.eval()
30
+ print("START TO GET QWEN")
31
  model_name = 'Qwen/Qwen3-1.7B'
32
  QWEN_TOKENIZER = AutoTokenizer.from_pretrained(model_name, device='auto')
33
  QWEN_TOKENIZER.pad_token_id = QWEN_TOKENIZER.eos_token_id
34
+ print("QWEN TOKENIZER LOADED")
35
+ QWEN_MODEL = AutoModelForCausalLM.from_pretrained(model_name, device='auto').half()
36
+ print("QWEN MODEL LOADED")
37
  MODELS_LOADED = True
38
  print("LOAD ENDED")
39
 
templates/base.html CHANGED
@@ -42,7 +42,7 @@
42
  </div>
43
 
44
  <div class="footer">
45
- <p>Visit our <a href="https://github.com/mengsifei/SimpleAES">GitHub</a></p>
46
  </div>
47
 
48
  <!-- Modals -->
 
42
  </div>
43
 
44
  <div class="footer">
45
+ <p>Visit our <a href="https://github.com/mengsifei/SimpleAES" target="_blank" rel="noopener noreferrer">GitHub</a></p>
46
  </div>
47
 
48
  <!-- Modals -->