Skip to content

Commit 062517e

Browse files
authored
Disable schedule GPU tests (#418)
* Format. * Fix usage examples. * Disable schedule GPU tests.
1 parent caa0187 commit 062517e

2 files changed

Lines changed: 13 additions & 2 deletions

File tree

.github/workflows/test_gpu.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,8 @@
11
name: GPU tests
22

33
on:
4-
schedule:
5-
- cron: "0 0 * * *"
4+
# schedule:
5+
# - cron: "0 0 * * *"
66
issue_comment:
77
types: [created]
88
workflow_dispatch:

download_x.py

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,11 @@
1+
from transformers import AutoModelForCausalLM, AutoTokenizer
2+
3+
model_id = "mistralai/Mixtral-8x7B-Instruct-v0.1"
4+
tokenizer = AutoTokenizer.from_pretrained(model_id)
5+
6+
while True:
7+
try:
8+
model = AutoModelForCausalLM.from_pretrained(model_id, resume_download=True)
9+
break
10+
except Exception:
11+
pass

0 commit comments

Comments
 (0)