blanchon commited on
Commit
67c72fe
·
1 Parent(s): 32b797c
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -1,5 +1,3 @@
1
- from typing import Any
2
-
3
  import gradio as gr
4
  import PIL
5
  import spaces
@@ -14,7 +12,7 @@ from transformers import AutoTokenizer, LlamaForCausalLM
14
  MODEL_PREFIX: str = "HiDream-ai"
15
  LLAMA_MODEL_NAME: str = "meta-llama/Meta-Llama-3.1-8B-Instruct"
16
  MODEL_PATH = "HiDream-ai/HiDream-I1-Dev"
17
- MODEL_CONFIGS: dict[str, Any] = {
18
  "guidance_scale": 0.0,
19
  "num_inference_steps": 28,
20
  "shift": 6.0,
 
 
 
1
  import gradio as gr
2
  import PIL
3
  import spaces
 
12
  MODEL_PREFIX: str = "HiDream-ai"
13
  LLAMA_MODEL_NAME: str = "meta-llama/Meta-Llama-3.1-8B-Instruct"
14
  MODEL_PATH = "HiDream-ai/HiDream-I1-Dev"
15
+ MODEL_CONFIGS = {
16
  "guidance_scale": 0.0,
17
  "num_inference_steps": 28,
18
  "shift": 6.0,