File tree Expand file tree Collapse file tree 2 files changed +5
-3
lines changed Expand file tree Collapse file tree 2 files changed +5
-3
lines changed Original file line number Diff line number Diff line change 47
47
48
48
- name : Run all (incl. slow) tests on GPU
49
49
env :
50
- HF_API_TOKEN : ${{ secrets.HF_API_TOKEN }}
50
+ HUGGING_FACE_HUB_TOKEN : ${{ secrets.HUGGING_FACE_HUB_TOKEN }}
51
51
run : |
52
52
python -m pytest -n 1 --max-worker-restart=0 --dist=loadfile -s tests/
Original file line number Diff line number Diff line change @@ -275,7 +275,8 @@ def test_ldm_text2img_fast(self):
275
275
@unittest .skipIf (torch_device == "cpu" , "Stable diffusion is supposed to run on GPU" )
276
276
def test_stable_diffusion (self ):
277
277
# make sure here that pndm scheduler skips prk
278
- sd_pipe = StableDiffusionPipeline .from_pretrained ("CompVis/stable-diffusion-v1-1" ).to (torch_device )
278
+ sd_pipe = StableDiffusionPipeline .from_pretrained ("CompVis/stable-diffusion-v1-1" , use_auth_token = True )
279
+ sd_pipe = sd_pipe .to (torch_device )
279
280
280
281
prompt = "A painting of a squirrel eating a burger"
281
282
generator = torch .Generator (device = torch_device ).manual_seed (0 )
@@ -295,7 +296,8 @@ def test_stable_diffusion(self):
295
296
@slow
296
297
@unittest .skipIf (torch_device == "cpu" , "Stable diffusion is supposed to run on GPU" )
297
298
def test_stable_diffusion_fast_ddim (self ):
298
- sd_pipe = StableDiffusionPipeline .from_pretrained ("CompVis/stable-diffusion-v1-1" ).to (torch_device )
299
+ sd_pipe = StableDiffusionPipeline .from_pretrained ("CompVis/stable-diffusion-v1-1" , use_auth_token = True )
300
+ sd_pipe = sd_pipe .to (torch_device )
299
301
300
302
scheduler = DDIMScheduler (
301
303
beta_start = 0.00085 ,
You can’t perform that action at this time.
0 commit comments