Liu Hong Yuan Tom commited on
Commit
811cd07
·
verified ·
1 Parent(s): 8c68ee4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -3,15 +3,13 @@ import numpy as np
3
  import random
4
  import spaces
5
  import torch
6
- import os
7
  from diffusers import DiffusionPipeline, FlowMatchEulerDiscreteScheduler
8
  from transformers import CLIPTextModel, CLIPTokenizer,T5EncoderModel, T5TokenizerFast
9
 
10
  dtype = torch.bfloat16
11
  device = "cuda" if torch.cuda.is_available() else "cpu"
12
- HF_TOKEN = os.environ["HF_TOKEN"]
13
 
14
- pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16, token=HF_TOKEN).to(device)
15
 
16
  MAX_SEED = np.iinfo(np.int32).max
17
  MAX_IMAGE_SIZE = 2048
 
3
  import random
4
  import spaces
5
  import torch
 
6
  from diffusers import DiffusionPipeline, FlowMatchEulerDiscreteScheduler
7
  from transformers import CLIPTextModel, CLIPTokenizer,T5EncoderModel, T5TokenizerFast
8
 
9
  dtype = torch.bfloat16
10
  device = "cuda" if torch.cuda.is_available() else "cpu"
 
11
 
12
+ pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16).to(device)
13
 
14
  MAX_SEED = np.iinfo(np.int32).max
15
  MAX_IMAGE_SIZE = 2048