artificiallover0 commited on
Commit
fc6b4b7
·
verified ·
1 Parent(s): 8680e55

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -2,12 +2,13 @@ import gradio as gr
2
  import numpy as np
3
  import random
4
  import spaces #[uncomment to use ZeroGPU]
5
- from diffusers import DiffusionPipeline
6
  import torch
7
  from huggingface_hub import login
8
  import os
9
  a=os.getenv('key_for_man_asshole')
10
  login(token=a )
 
11
 
12
  device = "cuda" if torch.cuda.is_available() else "cpu"
13
  model_repo_id = "black-forest-labs/FLUX.1-dev" # Replace to the model you would like to use
 
2
  import numpy as np
3
  import random
4
  import spaces #[uncomment to use ZeroGPU]
5
+ from diffusers import DiffusionPipeline , DPMSolverMultistepScheduler
6
  import torch
7
  from huggingface_hub import login
8
  import os
9
  a=os.getenv('key_for_man_asshole')
10
  login(token=a )
11
+ use_karras_sigmas=True
12
 
13
  device = "cuda" if torch.cuda.is_available() else "cpu"
14
  model_repo_id = "black-forest-labs/FLUX.1-dev" # Replace to the model you would like to use