Commit ece3ec42 authored by mashun1's avatar mashun1
Browse files

Update app_lcm.py

parent 20739a92
......@@ -137,7 +137,7 @@ if torch.cuda.is_available():
print("Loaded on Device!")
# speed-up T5
pipe.text_encoder.to_bettertransformer()
# pipe.text_encoder.to_bettertransformer()
if USE_TORCH_COMPILE:
pipe.transformer = torch.compile(pipe.transformer, mode="reduce-overhead", fullgraph=True)
......
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