Hugging Face's logo

benjamin-paine
/
champ

Diffusers
Safetensors
CHAMPPipeline
Model card Files Community
benjamin-paine commited on
Commit
a0b573b
•
1 Parent(s): ea5de63

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +13 -12
README.md CHANGED
@@ -32,23 +32,24 @@ Now, you can create the pipeline, automatically pulling the weights from this re
32
 
33
  ```py
34
  from champ import CHAMPPipeline
35
- pipeline = CHAMPPipeline.from_pretrained(
36
- "benjamin-paine/champ",
37
- torch_dtype=torch.float16,
38
- variant="fp16",
39
- device="cuda",
40
- ).to("cuda", dtype=torch.float16)
41
  ```
42
 
43
  Or, as a single file:
44
 
45
  ```py
46
- pipeline = CHAMPPipeline.from_pretrained(
47
- "benjamin-paine/champ",
48
- torch_dtype=torch.float16,
49
- variant="fp16",
50
- device="cuda",
51
- ).to("cuda", dtype=torch.float16)
 
52
  ```
53
 
54
  Follow this format for execution:
 
32
 
33
  ```py
34
  from champ import CHAMPPipeline
35
+ pipeline = CHAMPPipeline.from_pretrained(
36
+ "benjamin-paine/champ",
37
+ torch_dtype=torch.float16,
38
+ variant="fp16",
39
+ device="cuda",
40
+ ).to("cuda", dtype=torch.float16)
41
  ```
42
 
43
  Or, as a single file:
44
 
45
  ```py
46
+ from champ import CHAMPPipeline
47
+ pipeline = CHAMPPipeline.from_single_file(
48
+ "benjamin-paine/champ",
49
+ torch_dtype=torch.float16,
50
+ variant="fp16",
51
+ device="cuda",
52
+ ).to("cuda", dtype=torch.float16)
53
  ```
54
 
55
  Follow this format for execution: