Skip to content

Commit 9310035

Browse files
committed
update
1 parent 30f13ed commit 9310035

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

.github/workflows/nightly_tests.yml

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -356,6 +356,8 @@ jobs:
356356
config:
357357
- backend: "bitsandbytes"
358358
test_location: "bnb"
359+
- backend: "gguf"
360+
test_location: "gguf"
359361
runs-on:
360362
group: aws-g6e-xlarge-plus
361363
container:
@@ -519,4 +521,4 @@ jobs:
519521
# if: always()
520522
# run: |
521523
# pip install slack_sdk tabulate
522-
# python utils/log_reports.py >> $GITHUB_STEP_SUMMARY
524+
# python utils/log_reports.py >> $GITHUB_STEP_SUMMARY

tests/quantization/gguf/test_gguf.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -75,7 +75,7 @@ def test_gguf_linear_layers(self):
7575
if isinstance(module, torch.nn.Linear) and hasattr(module.weight, "quant_type"):
7676
assert module.weight.dtype == torch.uint8
7777

78-
def test_gguf_memory(self):
78+
def test_gguf_memory_usage(self):
7979
quantization_config = GGUFQuantizationConfig(compute_dtype=self.torch_dtype)
8080

8181
model = FluxTransformer2DModel.from_single_file(

0 commit comments

Comments
 (0)