Hugging Face's logo

Spaces:
ovi054
/
Z-Image-LORA
Running on Zero

App Files Community
1
ovi054 commited on
Commit
e30e3e0
·
verified ·
1 Parent(s): a4cd3f3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
16
 
17
 
18
  # Load the model pipeline
19
- pipe = ZImagePipeline.from_pretrained("Tongyi-MAI/Z-Image-Turbo", torch_dtype=dtype, low_cpu_mem_usage=False).to(device)
20
 
21
  torch.cuda.empty_cache()
22
 
 
16
 
17
 
18
  # Load the model pipeline
19
+ pipe = ZImagePipeline.from_pretrained("Tongyi-MAI/Z-Image-Turbo", torch_dtype=dtype).to(device)
20
 
21
  torch.cuda.empty_cache()
22