Spaces:

alfredplpl
/

sd3-with-LLM

Running on Zero

alfredplpl commited on Oct 22, 2024

Commit

17f6f0d

verified ·

1 Parent(s): 47b93ee

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -19,8 +19,9 @@ model = AutoModelForCausalLM.from_pretrained(
     device_map="cuda",
     torch_dtype=torch.bfloat16,
     trust_remote_code=True,
 )
-tokenizer = AutoTokenizer.from_pretrained("google/gemma-2-2b-jpn-it")
 upsampler = pipeline(
     "text-generation",
     model=model,

     device_map="cuda",
     torch_dtype=torch.bfloat16,
     trust_remote_code=True,
+    token=os.environ["TOKEN"]
 )
+tokenizer = AutoTokenizer.from_pretrained("google/gemma-2-2b-jpn-it", token=os.environ["TOKEN"])
 upsampler = pipeline(
     "text-generation",
     model=model,