prithivMLmods commited on
Commit
7f640fa
1 Parent(s): 58cba78

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +15 -0
README.md CHANGED
@@ -23,6 +23,21 @@ license: creativeml-openrail-m
23
 
24
  <Gallery />
25
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
26
 
27
  ## Trigger words
28
 
 
23
 
24
  <Gallery />
25
 
26
+ ## Setting Up
27
+ ```
28
+ import torch
29
+ from pipelines import DiffusionPipeline
30
+
31
+ base_model = "black-forest-labs/FLUX.1-dev"
32
+ pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
33
+
34
+ lora_repo = "prithivMLmods/Canopus-Pixar-3D-FluxDev-LoRA"
35
+ trigger_word = "pixar" # Leave trigger_word blank if not used.
36
+ pipe.load_lora_weights(lora_repo)
37
+
38
+ device = torch.device("cuda")
39
+ pipe.to(device)
40
+ ```
41
 
42
  ## Trigger words
43