benjamin-paine commited on
Commit
e9155e0
1 Parent(s): 2339d40

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +0 -2
README.md CHANGED
@@ -33,7 +33,6 @@ pipeline = AniPortraitPipeline.from_pretrained(
33
  "benjamin-paine/aniportrait",
34
  torch_dtype=torch.float16,
35
  variant="fp16",
36
- device="cuda"
37
  ).to("cuda", dtype=torch.float16)
38
  ```
39
 
@@ -45,7 +44,6 @@ pipeline = AniPortraitPipeline.from_single_file(
45
  "benjamin-paine/aniportrait",
46
  torch_dtype=torch.float16,
47
  variant="fp16",
48
- device="cuda"
49
  ).to("cuda", dtype=torch.float16)
50
  ```
51
 
 
33
  "benjamin-paine/aniportrait",
34
  torch_dtype=torch.float16,
35
  variant="fp16",
 
36
  ).to("cuda", dtype=torch.float16)
37
  ```
38
 
 
44
  "benjamin-paine/aniportrait",
45
  torch_dtype=torch.float16,
46
  variant="fp16",
 
47
  ).to("cuda", dtype=torch.float16)
48
  ```
49