Hugging Face's logo

benjamin-paine
/
champ

Diffusers
Safetensors
CHAMPPipeline
Model card Files Community
benjamin-paine commited on
Commit
7152faf
·
verified ·
1 Parent(s): a0b573b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -36,7 +36,7 @@ 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
 
@@ -48,7 +48,7 @@ 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
 
 
36
  "benjamin-paine/champ",
37
  torch_dtype=torch.float16,
38
  variant="fp16",
39
+ device="cuda"
40
  ).to("cuda", dtype=torch.float16)
41
  ```
42
 
 
48
  "benjamin-paine/champ",
49
  torch_dtype=torch.float16,
50
  variant="fp16",
51
+ device="cuda"
52
  ).to("cuda", dtype=torch.float16)
53
  ```
54