azhan77168 commited on
Commit
fd92bce
·
verified ·
1 Parent(s): dee7e77

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -1,4 +1,5 @@
1
  import spaces
 
2
  from peft import PeftModel
3
  import gradio as gr
4
  from diffusers import StableDiffusionImg2ImgPipeline
@@ -10,8 +11,10 @@ from diffusers import StableDiffusionPipeline
10
  # Load the model
11
  # model_id = "nitrosocke/Ghibli-Diffusion"
12
  # pipe = StableDiffusionImg2ImgPipeline.from_pretrained(model_id, torch_dtype=torch.float32)
 
 
13
  model_id = "black-forest-labs/FLUX.1-dev"
14
- pipe =DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.bfloat16,token=True)
15
 
16
 
17
  # # 1. 选择一个基础模型,例如 SD 1.5
 
1
  import spaces
2
+ import os
3
  from peft import PeftModel
4
  import gradio as gr
5
  from diffusers import StableDiffusionImg2ImgPipeline
 
11
  # Load the model
12
  # model_id = "nitrosocke/Ghibli-Diffusion"
13
  # pipe = StableDiffusionImg2ImgPipeline.from_pretrained(model_id, torch_dtype=torch.float32)
14
+ tk=os.getenv('ghtoken')
15
+ print("ttttt",tk)
16
  model_id = "black-forest-labs/FLUX.1-dev"
17
+ pipe =DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.bfloat16,token=tk)
18
 
19
 
20
  # # 1. 选择一个基础模型,例如 SD 1.5