Skip to content

Commit 5d3923b

Browse files
authored
Fix LCM benchmark test (#7158)
* make workflow dispatchable. * fix: lcm lora compile
1 parent 9451235 commit 5d3923b

File tree

2 files changed

+2
-0
lines changed

2 files changed

+2
-0
lines changed

Diff for: .github/workflows/benchmark.yml

+1
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
name: Benchmarking tests
22

33
on:
4+
workflow_dispatch:
45
schedule:
56
- cron: "30 1 1,15 * *" # every 2 weeks on the 1st and the 15th of every month at 1:30 AM
67

Diff for: benchmarks/base_classes.py

+1
Original file line numberDiff line numberDiff line change
@@ -141,6 +141,7 @@ def __init__(self, args):
141141
super().__init__(args)
142142
self.pipe.load_lora_weights(self.lora_id)
143143
self.pipe.fuse_lora()
144+
self.pipe.unload_lora_weights()
144145
self.pipe.scheduler = LCMScheduler.from_config(self.pipe.scheduler.config)
145146

146147
def get_result_filepath(self, args):

0 commit comments

Comments
 (0)