python 從huggingface中批量下載腳本2024-11-02

from huggingface_hub import snapshot_download
import time
import logging

# 設(shè)置日志配置
logging.basicConfig(level=logging.INFO)

max_retries = 15  # 最大重試次數(shù)
retries = 0
model_name = "h94/IP-Adapter"
local_name=r"E:\downloads\huggingface\ip_adapter"

while retries < max_retries:
    try:
        # 嘗試下載
        snapshot_download(
            repo_id=model_name,
            ignore_patterns=["*.bin"],  # 忽略下載哪些文件
            local_dir=local_name,
            force_download=False  # 默認(rèn)情況下不強(qiáng)制重新下載,除非設(shè)置為 True
        )
        logging.info("下載成功")
        break
    except Exception as e:
        # 捕獲其他異常
        retries += 1
        logging.error(f"第 {retries} 次嘗試失敗油额,其他錯(cuò)誤: {e}")

    if retries < max_retries:
        time.sleep(2)  # 等待 2 秒后重試
    else:
        logging.critical("下載失敗讲坎,超過(guò)最大重試次數(shù)")

有異常處理

from huggingface_hub import snapshot_download
import time
import logging
import requests

# 設(shè)置日志配置
logging.basicConfig(level=logging.INFO)

max_retries = 15  # 最大重試次數(shù)
retries = 0
model_name = "h94/IP-Adapter"
local_name=r"E:\downloads\huggingface\ip_adapter"

while retries < max_retries:
    try:
        # 嘗試下載
        snapshot_download(
            repo_id=model_name,
            ignore_patterns=["*.bin"],  # 忽略下載哪些文件
            local_dir=local_name,
            force_download=False  # 默認(rèn)情況下不強(qiáng)制重新下載束凑,除非設(shè)置為 True
        )
        logging.info("下載成功")
        break
    except requests.exceptions.RequestException as e:
        # 捕獲請(qǐng)求相關(guān)的錯(cuò)誤
        retries += 1
        logging.error(f"第 {retries} 次嘗試失敗,請(qǐng)求錯(cuò)誤: {e}")
    except Exception as e:
        # 捕獲其他異常
        retries += 1
        logging.error(f"第 {retries} 次嘗試失敗蹄殃,其他錯(cuò)誤: {e}")

    if retries < max_retries:
        time.sleep(2)  # 等待 2 秒后重試
    else:
        logging.critical("下載失敗挽铁,超過(guò)最大重試次數(shù)")
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
  • 序言:七十年代末,一起剝皮案震驚了整個(gè)濱河市退敦,隨后出現(xiàn)的幾起案子,更是在濱河造成了極大的恐慌蚣抗,老刑警劉巖侈百,帶你破解...
    沈念sama閱讀 211,639評(píng)論 6 492
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現(xiàn)場(chǎng)離奇詭異翰铡,居然都是意外死亡钝域,警方通過(guò)查閱死者的電腦和手機(jī),發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 90,277評(píng)論 3 385
  • 文/潘曉璐 我一進(jìn)店門锭魔,熙熙樓的掌柜王于貴愁眉苦臉地迎上來(lái)例证,“玉大人,你說(shuō)我怎么就攤上這事迷捧≈郑” “怎么了?”我有些...
    開(kāi)封第一講書人閱讀 157,221評(píng)論 0 348
  • 文/不壞的土叔 我叫張陵漠秋,是天一觀的道長(zhǎng)烦感。 經(jīng)常有香客問(wèn)我,道長(zhǎng)膛堤,這世上最難降的妖魔是什么? 我笑而不...
    開(kāi)封第一講書人閱讀 56,474評(píng)論 1 283
  • 正文 為了忘掉前任晌该,我火速辦了婚禮肥荔,結(jié)果婚禮上绿渣,老公的妹妹穿的比我還像新娘。我一直安慰自己燕耿,他們只是感情好中符,可當(dāng)我...
    茶點(diǎn)故事閱讀 65,570評(píng)論 6 386
  • 文/花漫 我一把揭開(kāi)白布。 她就那樣靜靜地躺著誉帅,像睡著了一般淀散。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上蚜锨,一...
    開(kāi)封第一講書人閱讀 49,816評(píng)論 1 290
  • 那天档插,我揣著相機(jī)與錄音,去河邊找鬼亚再。 笑死郭膛,一個(gè)胖子當(dāng)著我的面吹牛,可吹牛的內(nèi)容都是我干的氛悬。 我是一名探鬼主播则剃,決...
    沈念sama閱讀 38,957評(píng)論 3 408
  • 文/蒼蘭香墨 我猛地睜開(kāi)眼,長(zhǎng)吁一口氣:“原來(lái)是場(chǎng)噩夢(mèng)啊……” “哼如捅!你這毒婦竟也來(lái)了棍现?” 一聲冷哼從身側(cè)響起,我...
    開(kāi)封第一講書人閱讀 37,718評(píng)論 0 266
  • 序言:老撾萬(wàn)榮一對(duì)情侶失蹤镜遣,失蹤者是張志新(化名)和其女友劉穎己肮,沒(méi)想到半個(gè)月后,有當(dāng)?shù)厝嗽跇?shù)林里發(fā)現(xiàn)了一具尸體烈涮,經(jīng)...
    沈念sama閱讀 44,176評(píng)論 1 303
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡朴肺,尸身上長(zhǎng)有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 36,511評(píng)論 2 327
  • 正文 我和宋清朗相戀三年,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了坚洽。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片戈稿。...
    茶點(diǎn)故事閱讀 38,646評(píng)論 1 340
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡,死狀恐怖讶舰,靈堂內(nèi)的尸體忽然破棺而出鞍盗,到底是詐尸還是另有隱情,我是刑警寧澤跳昼,帶...
    沈念sama閱讀 34,322評(píng)論 4 330
  • 正文 年R本政府宣布般甲,位于F島的核電站告唆,受9級(jí)特大地震影響膘格,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜柴我,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 39,934評(píng)論 3 313
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望锚烦。 院中可真熱鬧觅闽,春花似錦、人聲如沸涮俄。這莊子的主人今日做“春日...
    開(kāi)封第一講書人閱讀 30,755評(píng)論 0 21
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽(yáng)彻亲。三九已至孕锄,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間苞尝,已是汗流浹背畸肆。 一陣腳步聲響...
    開(kāi)封第一講書人閱讀 31,987評(píng)論 1 266
  • 我被黑心中介騙來(lái)泰國(guó)打工, 沒(méi)想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留野来,地道東北人恼除。 一個(gè)月前我還...
    沈念sama閱讀 46,358評(píng)論 2 360
  • 正文 我出身青樓,卻偏偏與公主長(zhǎng)得像曼氛,于是被迫代替她去往敵國(guó)和親豁辉。 傳聞我的和親對(duì)象是個(gè)殘疾皇子,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 43,514評(píng)論 2 348

推薦閱讀更多精彩內(nèi)容