Spaces:
Running
on
Zero
Running
on
Zero
gokaygokay
commited on
Commit
•
2de338e
1
Parent(s):
c208235
Update app.py
Browse files
app.py
CHANGED
@@ -39,7 +39,6 @@ kolors_pipe = StableDiffusionXLPipeline(
|
|
39 |
scheduler=scheduler,
|
40 |
force_zeros_for_empty_prompt=False
|
41 |
).to(device)
|
42 |
-
kolors_pipe.enable_model_cpu_offload()
|
43 |
|
44 |
# VLM Captioner
|
45 |
vlm_model = PaliGemmaForConditionalGeneration.from_pretrained("gokaygokay/sd3-long-captioner-v2").to(device).eval()
|
|
|
39 |
scheduler=scheduler,
|
40 |
force_zeros_for_empty_prompt=False
|
41 |
).to(device)
|
|
|
42 |
|
43 |
# VLM Captioner
|
44 |
vlm_model = PaliGemmaForConditionalGeneration.from_pretrained("gokaygokay/sd3-long-captioner-v2").to(device).eval()
|