benjamin-paine commited on
Commit
dede4bf
·
verified ·
1 Parent(s): 21e34a0

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +11 -3
README.md CHANGED
@@ -87,18 +87,26 @@ pipeline = VidXTendPipeline.from_pretrained(
87
  "benjamin-paine/vidxtend",
88
  torch_dtype=torch.float16,
89
  variant="fp16",
90
- ).to("cuda", dtype=torch.float16)
91
  ```
92
 
93
  Or, as a single file:
94
 
95
  ```py
96
- from aniportrait import VidXTendPipeline
97
  pipeline = VidXTendPipeline.from_single_file(
98
  "benjamin-paine/vidxtend",
99
  torch_dtype=torch.float16,
100
  variant="fp16",
101
- ).to("cuda", dtype=torch.float16)
 
 
 
 
 
 
 
 
102
  ```
103
 
104
  Usage is as follows:
 
87
  "benjamin-paine/vidxtend",
88
  torch_dtype=torch.float16,
89
  variant="fp16",
90
+ )
91
  ```
92
 
93
  Or, as a single file:
94
 
95
  ```py
96
+ from vidxtend import VidXTendPipeline
97
  pipeline = VidXTendPipeline.from_single_file(
98
  "benjamin-paine/vidxtend",
99
  torch_dtype=torch.float16,
100
  variant="fp16",
101
+ )
102
+ ```
103
+
104
+ Use these methods to improve performance:
105
+
106
+ ```
107
+ pipeline.enable_model_cpu_offload()
108
+ pipeline.enable_vae_slicing()
109
+ pipeline.set_use_memory_efficient_attention_xformers()
110
  ```
111
 
112
  Usage is as follows: