Quellcode durchsuchen

Merge pull request #212 from zzzpppy/patch-2

增加HF_ENDPOINT,避免Connection aborted.
jjyaoao vor 5 Monaten
Ursprung
Commit
7064cd02cd
1 geänderte Dateien mit 4 neuen und 0 gelöschten Zeilen
  1. 4 0
      code/chapter3/Qwen.py

+ 4 - 0
code/chapter3/Qwen.py

@@ -1,3 +1,7 @@
+# 增加HF_ENDPOINT,避免Connection aborted. 
+import os
+os.environ["HF_ENDPOINT"] = "https://hf-mirror.com"
+
 import torch
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 from transformers import AutoModelForCausalLM, AutoTokenizer