feat: 初始提交

gpt2/infer.py

+from transformers import pipeline, set_seed
+generator = pipeline('text-generation', model='openai-community/gpt2', device='cuda')
+output = generator("Hello, I'm a language model,", truncation=True, max_length=30, num_return_sequences=1)[0][
+    "generated_text"]
+print(output)

hf_down.py

+# pip install huggingface-cli
+import os
+
+os.environ['HF_ENDPOINT'] = 'https://hf-mirror.com'
+
+model_list = [
+    "openai-community/gpt2",
+    # "stabilityai/stable-diffusion-xl-base-1.0"
+]
+
+os.system("pip install -U huggingface-hub")
+
+for model_path in model_list:
+    root_dir = os.path.basename(model_path)
+    os.system(
+        f"huggingface-cli download --resume-download  {model_path} --local-dir ./{root_dir}/{model_path} \
+        --local-dir-use-symlinks False --token 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")