benjamin-paine
/

champ

Model card Files Files and versions Community

benjamin-paine commited on Mar 26

Commit

7152faf

•

1 Parent(s): a0b573b

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -36,7 +36,7 @@ pipeline = CHAMPPipeline.from_pretrained(
   "benjamin-paine/champ",
   torch_dtype=torch.float16,
   variant="fp16",
-  device="cuda",
 ).to("cuda", dtype=torch.float16)
 ```
@@ -48,7 +48,7 @@ pipeline = CHAMPPipeline.from_single_file(
   "benjamin-paine/champ",
   torch_dtype=torch.float16,
   variant="fp16",
-  device="cuda",
 ).to("cuda", dtype=torch.float16)
 ```

   "benjamin-paine/champ",
   torch_dtype=torch.float16,
   variant="fp16",
+  device="cuda"
 ).to("cuda", dtype=torch.float16)
 ```
   "benjamin-paine/champ",
   torch_dtype=torch.float16,
   variant="fp16",
+  device="cuda"
 ).to("cuda", dtype=torch.float16)
 ```