Skip to content

GitLab

Commit e8de2628 authored by muyangli's avatar muyangli
Browse files

commented the hypersd lora

parent 7bc134d7
Show whitespace changes
Inline Side-by-side
Showing with 18 additions and 18 deletions
tests/flux/test_flux_dev_loras.py
View file @ e8de2628
...@@ -35,24 +35,24 @@ def test_flux_dev_loras(num_inference_steps, lora_name, lora_strength, cpu_offlo ...@@ -35,24 +35,24 @@ def test_flux_dev_loras(num_inference_steps, lora_name, lora_strength, cpu_offlo
) )
@pytest.mark.skipif(is_turing(), reason="Skip tests due to using Turing GPUs") # @pytest.mark.skipif(is_turing(), reason="Skip tests due to using Turing GPUs")
def test_flux_dev_hypersd8_1536x2048(): # def test_flux_dev_hypersd8_1536x2048():
run_test( # run_test(
precision=get_precision(), # precision=get_precision(),
model_name="flux.1-dev", # model_name="flux.1-dev",
dataset_name="MJHQ", # dataset_name="MJHQ",
height=1536, # height=1536,
width=2048, # width=2048,
num_inference_steps=8, # num_inference_steps=8,
guidance_scale=3.5, # guidance_scale=3.5,
use_qencoder=False, # use_qencoder=False,
attention_impl="nunchaku-fp16", # attention_impl="nunchaku-fp16",
cpu_offload=True, # cpu_offload=True,
lora_names="hypersd8", # lora_names="hypersd8",
lora_strengths=0.125, # lora_strengths=0.125,
cache_threshold=0, # cache_threshold=0,
expected_lpips=0.164, # expected_lpips=0.164,
) # )
@pytest.mark.skipif(is_turing(), reason="Skip tests due to using Turing GPUs") @pytest.mark.skipif(is_turing(), reason="Skip tests due to using Turing GPUs")
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment