Skip to content

GitLab

Unverified Commit 5d3923b6 authored by Sayak Paul's avatar Sayak Paul Committed by GitHub
Browse files

Fix LCM benchmark test (#7158) 

* make workflow dispatchable.

* fix: lcm lora compile
parent 9451235e
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
.github/workflows/benchmark.yml
View file @ 5d3923b6
name: Benchmarking tests name: Benchmarking tests
on: on:
workflow_dispatch:
schedule: schedule:
- cron: "30 1 1,15 * *" # every 2 weeks on the 1st and the 15th of every month at 1:30 AM - cron: "30 1 1,15 * *" # every 2 weeks on the 1st and the 15th of every month at 1:30 AM
......
benchmarks/base_classes.py
View file @ 5d3923b6
...@@ -141,6 +141,7 @@ class LCMLoRATextToImageBenchmark(TextToImageBenchmark): ...@@ -141,6 +141,7 @@ class LCMLoRATextToImageBenchmark(TextToImageBenchmark):
super().__init__(args) super().__init__(args)
self.pipe.load_lora_weights(self.lora_id) self.pipe.load_lora_weights(self.lora_id)
self.pipe.fuse_lora() self.pipe.fuse_lora()
self.pipe.unload_lora_weights()
self.pipe.scheduler = LCMScheduler.from_config(self.pipe.scheduler.config) self.pipe.scheduler = LCMScheduler.from_config(self.pipe.scheduler.config)
def get_result_filepath(self, args): def get_result_filepath(self, args):
......
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