Linoy Tsaban commited on
Commit
15bc341
1 Parent(s): 75e37c8

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -33,7 +33,7 @@ instance_prompt: a webpage in the style of <s0><s1>
33
  license: openrail++
34
  ---
35
 
36
- # SDXL LoRA DreamBooth - LinoyTsaban/web_y2k_v6
37
 
38
  <Gallery />
39
 
@@ -58,8 +58,8 @@ from huggingface_hub import hf_hub_download
58
  from safetensors.torch import load_file
59
 
60
  pipeline = AutoPipelineForText2Image.from_pretrained('stabilityai/stable-diffusion-xl-base-1.0', torch_dtype=torch.float16).to('cuda')
61
- pipeline.load_lora_weights('LinoyTsaban/web_y2k_v6', weight_name='pytorch_lora_weights.safetensors')
62
- embedding_path = hf_hub_download(repo_id='LinoyTsaban/web_y2k_v6', filename="embeddings.safetensors", repo_type="model")
63
  state_dict = load_file(embedding_path)
64
  pipeline.load_textual_inversion(state_dict["clip_l"], token=["<s0>", "<s1>"], text_encoder=pipe.text_encoder, tokenizer=pipe.tokenizer)
65
  pipeline.load_textual_inversion(state_dict["clip_g"], token=["<s0>", "<s1>"], text_encoder=pipe.text_encoder_2, tokenizer=pipe.tokenizer_2)
 
33
  license: openrail++
34
  ---
35
 
36
+ # SDXL LoRA DreamBooth - LinoyTsaban/web_y2k
37
 
38
  <Gallery />
39
 
 
58
  from safetensors.torch import load_file
59
 
60
  pipeline = AutoPipelineForText2Image.from_pretrained('stabilityai/stable-diffusion-xl-base-1.0', torch_dtype=torch.float16).to('cuda')
61
+ pipeline.load_lora_weights('LinoyTsaban/web_y2k', weight_name='pytorch_lora_weights.safetensors')
62
+ embedding_path = hf_hub_download(repo_id='LinoyTsaban/web_y2k', filename="embeddings.safetensors", repo_type="model")
63
  state_dict = load_file(embedding_path)
64
  pipeline.load_textual_inversion(state_dict["clip_l"], token=["<s0>", "<s1>"], text_encoder=pipe.text_encoder, tokenizer=pipe.tokenizer)
65
  pipeline.load_textual_inversion(state_dict["clip_g"], token=["<s0>", "<s1>"], text_encoder=pipe.text_encoder_2, tokenizer=pipe.tokenizer_2)