KHongJae commited on
Commit
25e9472
1 Parent(s): e4b21b6

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -2
README.md CHANGED
@@ -33,8 +33,7 @@ DreamBooth for the text encoder was enabled: True.
33
 
34
  ```python
35
  pipeline = DiffusionPipeline.from_pretrained(
36
- "KHongJae/Chatting_Based_Emoji_Generation_Model", torch_dtype=torch.float16, safety_checker=None
37
- ).to("cuda")
38
  pipeline.scheduler = UniPCMultistepScheduler.from_config(pipeline.scheduler.config)
39
  prompt = "Create your own prompt"
40
  negative_prompt = "Create your own negative prompt"
 
33
 
34
  ```python
35
  pipeline = DiffusionPipeline.from_pretrained(
36
+ "KHongJae/Chatting_Based_Emoji_Generation_Model", torch_dtype=torch.float16).to("cuda")
 
37
  pipeline.scheduler = UniPCMultistepScheduler.from_config(pipeline.scheduler.config)
38
  prompt = "Create your own prompt"
39
  negative_prompt = "Create your own negative prompt"