Hugging Face's logo

jokerbit
/
flux-qa-4

Model card Files Community
jokerbit commited on
Commit
b13975e
·
verified ·
1 Parent(s): 9aad41a

Fix cuda

Browse files
Files changed (1) hide show
  1. src/pipeline.py +1 -1
src/pipeline.py CHANGED
@@ -52,7 +52,7 @@ def load_pipeline() -> Pipeline:
52
  # pipeline.transformer.to(memory_format=torch.channels_last)
53
  # quantize_(pipeline.vae, int8_weight_only())
54
  # pipeline.vae = torch.compile(pipeline.vae, mode="reduce-overhead")
55
- # pipeline.to("cuda")
56
  for _ in range(2):
57
  pipeline("cat", num_inference_steps=4)
58
 
 
52
  # pipeline.transformer.to(memory_format=torch.channels_last)
53
  # quantize_(pipeline.vae, int8_weight_only())
54
  # pipeline.vae = torch.compile(pipeline.vae, mode="reduce-overhead")
55
+ pipeline.to("cuda")
56
  for _ in range(2):
57
  pipeline("cat", num_inference_steps=4)
58