try to debug qwen?
Browse files- create_app.py +4 -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 |
-
|
|
|
|
|
|
|
| 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 |
-
|
| 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 -->
|