Yuanshi commited on
Commit
f623d12
·
1 Parent(s): 9fbf1b0
Files changed (2) hide show
  1. ominicontrol.py +4 -4
  2. requirements.txt +9 -0
ominicontrol.py CHANGED
@@ -21,25 +21,25 @@ pipe.load_lora_weights(
21
  "Yuanshi/OminiControlStyle",
22
  weight_name=f"v0/ghibli.safetensors",
23
  adapter_name="ghibli",
24
- use_auth_token=HF_TOKEN
25
  )
26
  pipe.load_lora_weights(
27
  "Yuanshi/OminiControlStyle",
28
  weight_name=f"v0/irasutoya.safetensors",
29
  adapter_name="irasutoya",
30
- use_auth_token=HF_TOKEN
31
  )
32
  pipe.load_lora_weights(
33
  "Yuanshi/OminiControlStyle",
34
  weight_name=f"v0/simpsons.safetensors",
35
  adapter_name="simpsons",
36
- use_auth_token=HF_TOKEN
37
  )
38
  pipe.load_lora_weights(
39
  "Yuanshi/OminiControlStyle",
40
  weight_name=f"v0/snoopy.safetensors",
41
  adapter_name="snoopy",
42
- use_auth_token=HF_TOKEN
43
  )
44
 
45
 
 
21
  "Yuanshi/OminiControlStyle",
22
  weight_name=f"v0/ghibli.safetensors",
23
  adapter_name="ghibli",
24
+ token=HF_TOKEN
25
  )
26
  pipe.load_lora_weights(
27
  "Yuanshi/OminiControlStyle",
28
  weight_name=f"v0/irasutoya.safetensors",
29
  adapter_name="irasutoya",
30
+ token=HF_TOKEN
31
  )
32
  pipe.load_lora_weights(
33
  "Yuanshi/OminiControlStyle",
34
  weight_name=f"v0/simpsons.safetensors",
35
  adapter_name="simpsons",
36
+ token=HF_TOKEN
37
  )
38
  pipe.load_lora_weights(
39
  "Yuanshi/OminiControlStyle",
40
  weight_name=f"v0/snoopy.safetensors",
41
  adapter_name="snoopy",
42
+ token=HF_TOKEN
43
  )
44
 
45
 
requirements.txt ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ transformers
2
+ diffusers
3
+ peft
4
+ opencv-python
5
+ protobuf
6
+ sentencepiece
7
+ gradio
8
+ jupyter
9
+ torchao