Update README.md
Browse files
README.md
CHANGED
|
@@ -41,6 +41,7 @@ pipe = Flux2KleinPipeline.from_pretrained(
|
|
| 41 |
text_encoder=text_encoder_8bit,
|
| 42 |
torch_dtype=dtype,
|
| 43 |
)
|
|
|
|
| 44 |
pipe.enable_model_cpu_offload() # save some VRAM by offloading the model to CPU,GPU < 32G
|
| 45 |
|
| 46 |
prompt = "A cat holding a sign that says hello world"
|
|
|
|
| 41 |
text_encoder=text_encoder_8bit,
|
| 42 |
torch_dtype=dtype,
|
| 43 |
)
|
| 44 |
+
#pipe.to("cuda") # GPU < 32G
|
| 45 |
pipe.enable_model_cpu_offload() # save some VRAM by offloading the model to CPU,GPU < 32G
|
| 46 |
|
| 47 |
prompt = "A cat holding a sign that says hello world"
|