Spaces:

lea97338
/

Encoder

Paused

lea97338 commited on May 31

Commit

1580081

verified ·

1 Parent(s): 0ce502f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -10,6 +10,9 @@ dtype = torch.float32   # CPU-safe
 # On charge la pipeline mais on supprime tout sauf le text encoder
 pipe = Flux2Pipeline.from_pretrained(
     REPO_ID,
     torch_dtype=dtype,
     low_cpu_mem_usage=True,
 )

 # On charge la pipeline mais on supprime tout sauf le text encoder
 pipe = Flux2Pipeline.from_pretrained(
     REPO_ID,
+    transformer=None,
+    vae=None,
+    scheduler=None,
     torch_dtype=dtype,
     low_cpu_mem_usage=True,
 )