TastyRice commited on
Commit
8c1cb5c
1 Parent(s): aec8f3f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -10
app.py CHANGED
@@ -30,15 +30,6 @@ OUTPUT_DIR = os.getenv("OUTPUT_DIR", "./outputs")
30
 
31
  from diffusers.utils import make_image_grid
32
 
33
- ACCESS_TOKEN = os.environ["ACCESS_TOKEN"]
34
- login(token=ACCESS_TOKEN)
35
- pipeline = DiffusionPipeline.from_pretrained(
36
- "Tasty-Rice/Magic_on_paper/blob/main/Magic_on_paper-SDXL-v3.safetensors",
37
- trust_remote_code=True,
38
- torch_dtype=torch.float16,
39
- use_auth_token=ACCESS_TOKEN
40
- )
41
-
42
  MODEL = os.getenv(
43
  "MODEL",
44
  "https://huggingface.co/Tasty-Rice/Magic_on_paper/blob/main/Magic_on_paper-SDXL-v3.safetensors",
@@ -48,7 +39,6 @@ torch.backends.cudnn.deterministic = True
48
  torch.backends.cudnn.benchmark = False
49
 
50
  device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
51
- load_pipeline.to(device)
52
 
53
  def load_pipeline(model_name):
54
  vae = AutoencoderKL.from_pretrained(
 
30
 
31
  from diffusers.utils import make_image_grid
32
 
 
 
 
 
 
 
 
 
 
33
  MODEL = os.getenv(
34
  "MODEL",
35
  "https://huggingface.co/Tasty-Rice/Magic_on_paper/blob/main/Magic_on_paper-SDXL-v3.safetensors",
 
39
  torch.backends.cudnn.benchmark = False
40
 
41
  device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
 
42
 
43
  def load_pipeline(model_name):
44
  vae = AutoencoderKL.from_pretrained(