Lakonik commited on
Commit
f7a3995
·
1 Parent(s): 94d04f8

Add HF_TOKEN

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -7,15 +7,17 @@ import numpy as np
7
  import spaces
8
  import torch
9
 
 
 
 
10
  from lakonlab.models.architectures import OklabColorEncoder
11
  from lakonlab.models.diffusions.schedulers import FlowAdapterScheduler
12
  from lakonlab.pipelines.pipeline_pixelflux2_klein import PixelFlux2KleinPipeline
13
  from lakonlab.pipelines.prompt_rewriters.qwen3_vl import Qwen3VLPromptRewriter
14
  from lakonlab.ui.gradio.create_text_to_img import create_interface_text_to_img
15
 
16
-
17
- torch.backends.cuda.matmul.allow_tf32 = True
18
- torch.backends.cudnn.allow_tf32 = True
19
 
20
 
21
  DEFAULT_PROMPT = (
 
7
  import spaces
8
  import torch
9
 
10
+ torch.backends.cuda.matmul.allow_tf32 = True
11
+ torch.backends.cudnn.allow_tf32 = True
12
+
13
  from lakonlab.models.architectures import OklabColorEncoder
14
  from lakonlab.models.diffusions.schedulers import FlowAdapterScheduler
15
  from lakonlab.pipelines.pipeline_pixelflux2_klein import PixelFlux2KleinPipeline
16
  from lakonlab.pipelines.prompt_rewriters.qwen3_vl import Qwen3VLPromptRewriter
17
  from lakonlab.ui.gradio.create_text_to_img import create_interface_text_to_img
18
 
19
+ from huggingface_hub import login
20
+ login(token=os.getenv('HF_TOKEN'))
 
21
 
22
 
23
  DEFAULT_PROMPT = (