jn1xia commited on
Commit
0310844
·
verified ·
1 Parent(s): 83c5a9d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -10,6 +10,9 @@ import torch
10
  from accelerate import Accelerator
11
  #model_id = "CompVis/stable-diffusion-v1-4"
12
  #pipe = StableDiffusionPipeline.from_pretrained(model_id)
 
 
 
13
 
14
  # Ensure the model is using the CPU
15
  #pipe = pipe.to("cpu")
@@ -32,7 +35,7 @@ def generate_images_using_huggingface_diffusers(text):
32
  #pipe.load_lora_weights("strangerzonehf/Flux-Midjourney-Mix2-LoRA")
33
 
34
  pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev")
35
- pipe.load_lora_weights("xey/sldr_flux_nsfw_v2-studio")
36
 
37
 
38
  # pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", torch_dtype=torch.float32)
 
10
  from accelerate import Accelerator
11
  #model_id = "CompVis/stable-diffusion-v1-4"
12
  #pipe = StableDiffusionPipeline.from_pretrained(model_id)
13
+
14
+ from huggingface_hub import login
15
+ login(token="your_access_token")
16
 
17
  # Ensure the model is using the CPU
18
  #pipe = pipe.to("cpu")
 
35
  #pipe.load_lora_weights("strangerzonehf/Flux-Midjourney-Mix2-LoRA")
36
 
37
  pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev")
38
+ #pipe.load_lora_weights("xey/sldr_flux_nsfw_v2-studio")
39
 
40
 
41
  # pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", torch_dtype=torch.float32)